Camera Calibration in Python

Understanding Camera Calibration

Camera calibration is the process of determining the intrinsic and extrinsic parameters of a camera. The intrinsic parameters relate to the camera's optics and sensor, including focal length, optical center, and lens distortion. Extrinsic parameters deal with the camera's position and orientation in the 3D world.

Think of calibration as a way to correct the various distortions and errors that occur when capturing images. Without proper calibration, pictures may appear warped, and measurements taken from these images could be inaccurate. This is particularly critical in applications such as 3D reconstruction, robotic vision, and augmented reality.

Why is Camera Calibration Important?

1. Distortion Correction: Cameras often introduce barrel or pincushion distortion. Calibration helps in rectifying these distortions.
2. Accurate Measurements: In applications like robotics, accurate distance measurements are crucial. Calibration ensures these measurements are reliable.
3. Image Rectification: It allows images taken from different cameras or viewpoints to be aligned properly, which is essential for stereo vision.

Getting Started with Camera Calibration in Python

To implement camera calibration, we’ll utilize the popular OpenCV library in Python. If you haven’t installed OpenCV yet, you can do so using pip:

pip install opencv-python

Step 1: Capture Calibration Images

To start with camera calibration, you need several images of a calibration pattern taken from different angles and distances. A commonly used pattern is a checkerboard. You can print a checkerboard pattern and capture images of it, or use an existing dataset.

Step 2: Find Chessboard Corners

You can use the cv2.findChessboardCorners function, which helps identify the corners of the checkerboard squares. Here's how you can do this in your code:

import cv2
import numpy as np
import glob

# Prepare object points
pattern_size = (9, 6)

# Inside corners of the checkerboard
objp = np.zeros((pattern_size[0] * pattern_size[1], 3), np.float32)
objp[:, :2] = np.mgrid[0:pattern_size[0], 0:pattern_size[1]].T.reshape(-1, 2)

# Arrays to store object points and image points
objpoints = []

# 3d point in real world space
imgpoints = []

# 2d points in image plane

# Load images
images = glob.glob('calibration/*.jpg')

# Adjust your path

for image_file in images:
    img = cv2.imread(image_file)
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# Find the chessboard corners
    ret, corners = cv2.findChessboardCorners(gray, pattern_size, None)
    
    if ret:
        objpoints.append(objp)
        imgpoints.append(corners)
        cv2.drawChessboardCorners(img, pattern_size, corners, ret)
        cv2.imshow('Image', img)
        cv2.waitKey(500)

cv2.destroyAllWindows()

Step 3: Calibrate the Camera

Once you've collected your object points and image points, you can proceed to calibrate the camera. The cv2.calibrateCamera function does all the heavy lifting:

# Calibrate the camera
ret, mtx, dist, rvecs, tvecs = cv2.calibrateCamera(objpoints, imgpoints, gray.shape[::-1], None, None)

# Save the camera matrix and distortion coefficients
np.savez('calibration_parameters.npz', mtx=mtx, dist=dist)

Step 4: Undistort an Image

Now that you have the calibration parameters, you can undistort any image taken by the camera:

# Load the calibration parameters
with np.load('calibration_parameters.npz') as X:
    mtx, dist = [X[i] for i in ('mtx', 'dist')]

# Load an image to undistort
img = cv2.imread('calibration/test_image.jpg')
h, w = img.shape[:2]
newCamMtrx, roi = cv2.getOptimalNewCameraMatrix(mtx, dist, (w, h), 1, (w, h))

# Undistort
dst = cv2.undistort(img, mtx, dist, None, newCamMtrx)
x, y, w, h = roi
dst = dst[y:y+h, x:x+w]

# Crop the image

cv2.imshow('Undistorted Image', dst)
cv2.waitKey(0)
cv2.destroyAllWindows()

Conclusion

Camera calibration is a fundamental skill for anyone working in computer vision. By adequately calibrating your camera, you enhance the accuracy of your measurements and make your computer vision applications much more reliable. Through the use of OpenCV in Python, you can quickly implement calibration techniques and begin working on exciting computer vision projects. Keep experimenting and exploring more features that OpenCV has to offer!