DyM Podcast

As i promised you can use this code to implement the camera calibration:

# Import required modules

import cv2

import numpy as np

import os

import glob





# Define the dimensions of checkerboard

CHECKERBOARD = (8, 13)





# stop the iteration when specified

# accuracy, epsilon, is reached or

# specified number of iterations are completed.

criteria = (cv2.TERM_CRITERIA_EPS +

			cv2.TERM_CRITERIA_MAX_ITER, 30, 0.001)





# Vector for 3D points

threedpoints = []



# Vector for 2D points

twodpoints = []





# 3D points real world coordinates

objectp3d = np.zeros((1, CHECKERBOARD[0]

					* CHECKERBOARD[1],

					3), np.float32)

objectp3d[0, :, :2] = np.mgrid[0:CHECKERBOARD[0],

							0:CHECKERBOARD[1]].T.reshape(-1, 2)

prev_img_shape = None





# Extracting path of individual image stored

# in a given directory. Since no path is

# specified, it will take current directory

# jpg files alone

images = glob.glob(r'C:\Users\49179\PycharmProjects\face_recognitionv2\calibration\*.jpg')



for filename in images:

	image = cv2.imread(filename)

	grayColor = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)



	# Find the chess board corners

	# If desired number of corners are

	# found in the image then ret = true

	ret, corners = cv2.findChessboardCorners(

					grayColor, CHECKERBOARD,

					cv2.CALIB_CB_ADAPTIVE_THRESH

					+ cv2.CALIB_CB_FAST_CHECK +

					cv2.CALIB_CB_NORMALIZE_IMAGE)



	# If desired number of corners can be detected then,

	# refine the pixel coordinates and display

	# them on the images of checker board

	if ret == True:

		threedpoints.append(objectp3d)



		# Refining pixel coordinates

		# for given 2d points.

		corners2 = cv2.cornerSubPix(

		   grayColor, corners, (11, 11), (-1, -1), criteria)



		twodpoints.append(corners2)



		# Draw and display the corners

		image = cv2.drawChessboardCorners(image,

										CHECKERBOARD,

										corners2, ret)



		h, w = image.shape[:2]

		ret, matrix, distortion, r_vecs, t_vecs = cv2.calibrateCamera(threedpoints, twodpoints, grayColor.shape[::-1], None, None)

		print(" Camera matrix:")

		print(matrix)



		print("\n Distortion coefficient:")

		print(distortion)



		print("\n Rotation Vectors:")

		print(r_vecs)



		print("\n Translation Vectors:")

		print(t_vecs)

	cv2.imshow('img', image)



	cv2.waitKey(0)



cv2.destroyAllWindows()

Note: Its important to define the directory path correctly in order to run the code. Do not use Tab and space together to order the code. This makes problem to run the code in pycharm, only use Tab or Space