基于python opencv实现广角相机标定和图像畸变矫正
生活随笔
收集整理的這篇文章主要介紹了
基于python opencv实现广角相机标定和图像畸变矫正
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
目的:
代碼:
import cv2 as cv import numpy as np import glob import os#循環(huán)中斷 criteria=(cv.TERM_CRITERIA_EPS+cv.TERM_CRITERIA_MAX_ITER,30,0.001)#標(biāo)定板交叉點(diǎn)的個數(shù) row=6 column=4 objp=np.zeros((row*column,3),np.float32) objp[:,:2]=np.mgrid[0:row,0:column].T.reshape(-1,2)objpoints=[] #實際空間3D點(diǎn) imgpoints=[] #圖像中2D點(diǎn)#批量讀取圖片 images=glob.glob('D:\\python\\photos\\chess\\*.jpg') for fname in images:img=cv.imread(fname)gray=cv.cvtColor(img,cv.COLOR_BGR2GRAY)#找標(biāo)定板角點(diǎn)ret,corners=cv.findChessboardCorners(gray,(row,column),None)if ret==True:objpoints.append(objp)corners2=cv.cornerSubPix(gray, corners, (11,11), (-1,-1), criteria)imgpoints.append(corners2)#標(biāo)定相機(jī) ret,Matrix,dist,rvecs,tvecs=cv.calibrateCamera(objpoints,imgpoints,gray.shape[::-1],None,None)datadir="D:\\python\\photos\\chess\\"path=os.path.join(datadir) img_list=os.listdir(path)for i in img_list:img=cv.imread(os.path.join(path,i))h,w=img.shape[:2] newMatrix, roi = cv.getOptimalNewCameraMatrix(Matrix, dist, (w,h), 1, (w,h)) #矯正圖像dst = cv.undistort(img, Matrix, dist, None, newMatrix)cv.imwrite('D:\\python\\practice\\photos\\calibrated\\'+i,dst)#計算重投影誤差 tot_error = 0 for i in range(len(objpoints)):imgpoints2, _ = cv.projectPoints(objpoints[i], rvecs[i], tvecs[i], Matrix, dist)error = cv.norm(imgpoints[i],imgpoints2, cv.NORM_L2)/len(imgpoints2)tot_error += error#輸出參數(shù) print('ret:\n',ret) print('mtx:\n',Matrix) print('dist:\n',dist) print('rvecs:\n',rvecs) print('tvecs:\n',tvecs) print ("total error: ", tot_error/len(objpoints))歡迎關(guān)注公眾號:算法工程師的學(xué)習(xí)日志
總結(jié)
以上是生活随笔為你收集整理的基于python opencv实现广角相机标定和图像畸变矫正的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: python3 bytes和str转换,
- 下一篇: OpenCV-Python 相机校准和消