图像批量去畸变

import cv2
import numpy as np
import os


#相机内参
camera = np.array([[1254.8, 0, 901.533],
                   [0, 1261.2, 586.74],
                   [0, 0, 1]])
#相机畸变系数
dist = np.array([-0.384, 0.2382 ,-0.1157 ,-0.0018 ,0.0027])


path = '/home/wanghe/桌面/tunel(按照老要求标注)/'

for i in os.listdir(path):
    image = cv2.imread(path + i)
    image_undistort = cv2.undistort(image, camera, dist, None, camera) #去畸变
    cv2.imwrite('/home/wanghe/桌面/tunel/' + i,image_undistort)