BM算法求解视差图
目的
用BM算法计算下列两幅图的视差图。


代码
import numpy as np
import cv2
mum_disp = 16
#读取左图并转为单通道灰度图
imgL = cv2.imread('lzq1.jpg')
imgLG = cv2.cvtColor(imgL, cv2.COLOR_BGR2GRAY)
#读取右图并转为单通道灰度图
imgR = cv2.imread('lzq2.jpg')
imgRG = cv2.cvtColor(imgR, cv2.COLOR_BGR2GRAY)
#计算视差
stereo = cv2.StereoBM_create(numDisparities=mum_disp, blockSize=9)
disp = stereo.compute(imgLG, imgRG)
#输出图片
cv2.imwrite('disp.jpg', disp)
cv2.waitKey(0)
cv2.destroyAllWindows()
结果
