python_图像去畸变/图像矫正

说明:

• 输入:畸变图像
• 输出: 去畸变图像

代码:

import numpy as np
from cv2 import cv2 as cv

img_rgb = "1.jpg"

R = np.eye(3)
img_size = (1920, 1080)
camera_matrix = np.array( [788.82312, 0, 987.54279, 0, 787.83112, 577.85284, 0, 0, 1, ]).reshape([3, 3])
distortion_coefficients = np.array( [-0.025751632, -0.017036539, 0.01532208, -0.0057064975]).reshape([4, 1])
mapx, mapy = cv.fisheye.initUndistortRectifyMap( camera_matrix, distortion_coefficients, R, camera_matrix, img_size, cv.CV_32FC1)
srcImg = cv.imread(img_rgb)
# srcImg = cv.imread("camera.bmp")
resultImg = cv.remap(srcImg, mapx, mapy, cv.INTER_LINEAR,  cv.BORDER_CONSTANT)
cv.imwrite("result_1.png", resultImg)

结果: