Correction des lentilles fish-eye

Correction des lentilles fish-eye

introduction

Le drone utilisé dans la recherche étant une lentille fish-eye, il est devenu nécessaire de corriger l'image acquise.

D'après la référence [1], la correction générale semble être une "correction rectiligne", et il semble que cela puisse être fait avec opencv, alors je l'ai touchée.

Environnement

Le module opencv de python cv2 n'a pas de module d'étalonnage d'objectif fisheye

Par conséquent, construisez un environnement de ʻopencv c ++ `.

Mais il y avait une personne qui faisait ça avec ʻopen cv3 series`, alors je l'ai fait là-bas. Voir [4]

Copiez le motif en damier requis pour l'étalonnage avec A4.

チェッカーボード-1024x717.png

Prenez un motif en damier sous différents angles avec une caméra fish-eye (environ 20 clichés).

Courir

calibration.py


import cv2
# assert cv2.__version__[0] == '3', 'The fisheye module requires opencv version >= 3.0.0'
import numpy as np
import os
import glob

CHECKERBOARD = (6,9)

print(CHECKERBOARD)
subpix_criteria = (cv2.TERM_CRITERIA_EPS+cv2.TERM_CRITERIA_MAX_ITER, 30, 0.1)
calibration_flags = cv2.fisheye.CALIB_RECOMPUTE_EXTRINSIC+cv2.fisheye.CALIB_CHECK_COND+cv2.fisheye.CALIB_FIX_SKEW
objp = np.zeros((1, CHECKERBOARD[0]*CHECKERBOARD[1], 3), np.float32)
objp[0,:,:2] = np.mgrid[0:CHECKERBOARD[0], 0:CHECKERBOARD[1]].T.reshape(-1, 2)
_img_shape = None
objpoints = [] # 3d point in real world space
imgpoints = [] # 2d points in image plane.
images = glob.glob('source/*.jpg')
for fname in images:
    img = cv2.imread(fname)
    if _img_shape == None:
        _img_shape = img.shape[:2]
    else:
        assert _img_shape == img.shape[:2], "All images must share the same size."
    gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
    # Find the chess board corners
    ret, corners = cv2.findChessboardCorners(gray, CHECKERBOARD, cv2.CALIB_CB_ADAPTIVE_THRESH+cv2.CALIB_CB_FAST_CHECK+cv2.CALIB_CB_NORMALIZE_IMAGE)
    # If found, add object points, image points (after refining them)
    if ret == True:
        objpoints.append(objp)
        cv2.cornerSubPix(gray,corners,(3,3),(-1,-1),subpix_criteria)
        imgpoints.append(corners)

print('img end')
N_OK = len(objpoints)
K = np.zeros((3, 3))
D = np.zeros((4, 1))
rvecs = [np.zeros((1, 1, 3), dtype=np.float64) for i in range(N_OK)]
tvecs = [np.zeros((1, 1, 3), dtype=np.float64) for i in range(N_OK)]
rms, _, _, _, _ = \
    cv2.fisheye.calibrate(
        objpoints,
        imgpoints,
        gray.shape[::-1],
        K,
        D,
        rvecs,
        tvecs,
        calibration_flags,
        (cv2.TERM_CRITERIA_EPS+cv2.TERM_CRITERIA_MAX_ITER, 30, 1e-6)
    )
print("Found " + str(N_OK) + " valid images for calibration")
print("DIM=" + str(_img_shape[::-1]))
print("K=np.array(" + str(K.tolist()) + ")")
print("D=np.array(" + str(D.tolist()) + ")")

Lors de l'exécution, trois paramètres sont émis. DIM: Angle de vue K: coefficient de déformation radiale D: coefficient de distorsion circonférentielle

undistort.py


import numpy as np
import cv2
import sys
import os

# You should replace these 3 lines with the output in calibration step
DIM=##
K=##
D=##

def undistort(img_path):
    img = cv2.imread(img_path)
    h,w = img.shape[:2]
    map1, map2 = cv2.fisheye.initUndistortRectifyMap(K, D, np.eye(3), K, DIM, cv2.CV_16SC2)
    undistorted_img = cv2.remap(img, map1, map2, interpolation=cv2.INTER_LINEAR, borderMode=cv2.BORDER_CONSTANT)
    # cv2.imshow("undistorted", undistorted_img)
    # cv2.waitKey(0)
    # cv2.destroyAllWindows()
    cv2.imwrite('out_' + os.path.basename(img_path), undistorted_img)
if __name__ == '__main__':
    for p in sys.argv[1:]:
        undistort(p)

Exemple d'exécution

Bebop2_20201027162227+0900.jpg

out_Bebop2_20201027162227+0900.jpg

référence

  1. Si vous détectez un objet avec une caméra fish-eye, avez-vous une correction de distorsion? --Qiita

  2. GitHub - opencv/opencv: Open Source Computer Vision Library

  3. Exécutez C ++ OpenCV sur mac. --Qiita

  4. Calibrate fisheye lens using OpenCV — part 1 | by Kenneth Jiang | Medium

Recommended Posts

Correction des lentilles fish-eye
Essayez d'ajouter la distorsion de l'objectif fisheye à l'image