我想做一些与this question类似的事情,但是对于stereoCalibrate()
而不是calibrateCamera()
。也就是说,计算立体相机校准的重投影误差。
我的简化示例如下所示:
import cv2
import numpy as np
def calibrate_stereo(w, h, objpoints, imgpoints_l, imgpoints_r):
stereocalib_criteria = (cv2.TERM_CRITERIA_COUNT + cv2.TERM_CRITERIA_EPS , 1000, 1e-6)
retval, A1, D1, A2, D2, R, T, E, F = cv2.stereoCalibrate(objpoints,imgpoints_l, imgpoints_r,None,None,None,None, (w,h), flags=0, criteria=stereocalib_criteria)
return (retval, (A1,D1,A2,D2, R, T, E, F))
def calc_rms_stereo(objectpoints, imgpoints_l, imgpoints_r, A1, D1, A2, D2, R, T):
tot_error = 0
total_points = 0
for i, objpoints in enumerate(objectpoints):
# calculate world <-> cam1 transformation
_, rvec_l, tvec_l,_ = cv2.solvePnPRansac(objpoints, imgpoints_l[i], A1, D1)
# compute reprojection error for cam1
rp_l, _ = cv2.projectPoints(objpoints, rvec_l, tvec_l, A1, D1)
tot_error += np.sum(np.square(np.float64(imgpoints_l[i] - rp_l)))
total_points += len(objpoints)
# calculate world <-> cam2 transformation
rvec_r, tvec_r = cv2.composeRT(rvec_l,tvec_l,cv2.Rodrigues(R)[0],T)[:2]
# compute reprojection error for cam2
rp_r,_ = cv2.projectPoints(objpoints, rvec_r, tvec_r, A2, D2)
tot_error += np.square(imgpoints_r[i] - rp_r).sum()
total_points += len(objpoints)
mean_error = np.sqrt(tot_error/total_points)
return mean_error
if __name__ == "__main__":
# omitted: reading values for w,h, objectPoints, imgpoints_l, imgpoints_r from file (format as expected by the OpenCV functions)
# [...]
rms, (A1,D1,A2,D2,R,T,_,_) = calibrate_stereo(w, h, objectpoints, imgpoints_l, imgpoints_r)
print("RMS (stereo calib): {}".format(rms))
rms_2 = calc_rms_stereo(objectpoints, imgpoints_l, imgpoints_r, A1, D1, A2, D2, R, T)
print("RMS (custom calculation):", rms_2)
示例输出:
RMS (stereo calib): 0.14342257926694932
RMS (custom calculation): 0.356273345751
据我所知,stereoCalibrate()
的源代码中的计算与我的非常相似。我错过了什么?
Ubuntu上的OpenCV 3.3.0
答案 0 :(得分:0)
在实施基于OpenCV实现的自定义立体声校准算法后,我解决了这个问题。
cv2.stereoCalibrate()
内计算的重投影误差与我的自定义计算之间的差异源于外部参数rvec_l
和tvec_l
的不同值。这些矢量描述左相机和每个图像的校准图案之间的旋转和平移。 cv2.solvePnpRansac()
仅根据左图像的重投影误差生成优化值,而在cv2.stereoCalibrate()
中,基于两者中的重投影误差,这些值与R
和T
一起优化每个立体声对的图像。
如果想要完全复制cv2.stereoCalibrate()
返回的RMS值,则必须修改cv::stereoCalibrate()
的C / C ++源代码以返回优化的外部参数({{1那已经用于单目校准了。)