如何使用打开CV.omnidir鱼眼图像脱蜡模块

Question

我尝试使用omnidirectional module在Python中对鱼眼图像进行脱蜡处理。我试图在Python中修改这个C++ tutorial，但是遇到了一些问题。这是我的代码：

import numpy as np
import cv2
import glob

nx = 9
ny = 6

# termination criteria
criteria = (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 30, 0.001)

chessboard_model = np.zeros((1, nx*ny, 3), np.float32)
chessboard_model[0, :, :2] = np.mgrid[0:nx, 0:ny].T.reshape(-1, 2)

# prepare object points, like (0,0,0), (1,0,0), (2,0,0) ....,(6,5,0)
objp = np.zeros((nx*ny, 3), np.float32)
objp[:,:2] = np.mgrid[0:nx, 0:ny].T.reshape(-1,2)

# Arrays to store object points and image points from all the images.
objpoints = [] # 3d point in real world space
imgpoints = [] # 2d points in image plane.

filenames = glob.glob('/home/work/Downloads/fisheye_calibration/*.jpg')
images = []
i = 0
for fname in filenames:
    i += 1
    print("Processing image: {}".format(i))
    img = cv2.imread(fname)
    gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
    images.append(img)

    # Find the chess board corners
    ret, corners = cv2.findChessboardCorners(gray, (nx,ny),None)
    # If found, add object points, image points (after refining them)
    if ret == True:
        objpoints.append(objp)
        corners2 = cv2.cornerSubPix(gray,corners,(11,11),(-1,-1),criteria)
        #imgpoints.append(corners2)
        imgpoints.append(corners2.reshape(1, -1, 2))

        # Draw and display the corners
        img = cv2.drawChessboardCorners(img, (nx,ny), corners2,ret)
        cv2.imshow('img',img)
        cv2.waitKey(500)

num_points = len(imgpoints)
K = np.zeros((3, 3))
xi = np.array([])
idx = np.array([])
D = np.zeros((4, 1))
h, w = images[0].shape[:2]
calibration_flags = cv2.omnidir.CALIB_USE_GUESS | cv2.omnidir.CALIB_FIX_SKEW
obj_points_arr = np.array([chessboard_model]*num_points)
img_points_arr = np.array(imgpoints).squeeze(axis=1)
rms = cv2.omnidir.calibrate(obj_points_arr, img_points_arr, (w,h), K, xi, D, calibration_flags, criteria)

我得到以下错误：

^{pr2}$

这里obj_points_arr和{}分别是{}和{}。如果我使用

obj_points_arr = np.array([chessboard_model]*num_points).squeeze(axis=1)
img_points_arr = np.array(imgpoints).squeeze(axis=1)

然后我得到以下错误

cv2.error: /home/work/opencv/modules/core/src/matrix.cpp:469: error: (-215) 0 <= _rowRange.start && _rowRange.start <= _rowRange.end && _rowRange.end <= m.rows in function Mat

这些阵列的形状应该是什么？有人能分享他们用来去除鱼眼图像的python代码吗？在

Answer 1

首先你没有得到图像的角落，最好写信给其他人打印出来。在

else: print "not found: "+fname

你可以使用pyfisheye模块来完成你的任务。 https://bitbucket.org/amitibo/pyfisheye