undistortion.py 文件源码

def __init__(self, image_points, dest_points, size=(3840, 2160)):
        dest_3dpoints = [[x, y, 0] for x, y in dest_points]
        _, camera_matrix, dist_coeffs, _, _ = cv2.calibrateCamera(
                [np.float32([dest_3dpoints])],
                [np.float32([image_points])],
                size, None, None, flags=self.flags)

        self.image_size = size
        self.camera_matrix = camera_matrix
        self.dist_coeffs = dist_coeffs
        self.new_camera_matrix = cv2.getOptimalNewCameraMatrix(
                camera_matrix, dist_coeffs, self.image_size, 0)[0]