2019-04-29 09:50:26 +08:00
|
|
|
|
|
|
|
|
import numpy as np
|
|
|
|
|
import matplotlib
|
|
|
|
|
import cv2
|
|
|
|
|
import os
|
|
|
|
|
from matplotlib import pyplot as plt
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2019-04-29 18:09:11 +08:00
|
|
|
img = cv2.imread("judy.png", 0)
|
2019-04-29 09:50:26 +08:00
|
|
|
|
|
|
|
|
imgL = np.split(img, 2, 1)[0]
|
|
|
|
|
imgR = np.split(img, 2, 1)[1]
|
2019-04-29 14:10:03 +08:00
|
|
|
stereo = cv2.StereoBM_create(numDisparities=96, blockSize=11)
|
2019-04-29 09:50:26 +08:00
|
|
|
disparity = stereo.compute(imgL, imgR)
|
2019-04-29 14:10:03 +08:00
|
|
|
plt.title("BM")
|
2019-04-29 09:50:26 +08:00
|
|
|
plt.imshow(disparity)
|
|
|
|
|
plt.show()
|
