StereoVidComfort/StereoBM.py
2019-04-29 18:09:11 +08:00

19 lines
342 B
Python

import numpy as np
import matplotlib
import cv2
import os
from matplotlib import pyplot as plt
img = cv2.imread("judy.png", 0)
imgL = np.split(img, 2, 1)[0]
imgR = np.split(img, 2, 1)[1]
stereo = cv2.StereoBM_create(numDisparities=96, blockSize=11)
disparity = stereo.compute(imgL, imgR)
plt.title("BM")
plt.imshow(disparity)
plt.show()