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:04:53 +08:00
|
|
|
img = cv2.imread("bridge.png")
|
|
|
|
|
cv2.imshow("img", img)
|
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 18:04:53 +08:00
|
|
|
stereo = cv2.StereoSGBM_create(numDisparities=96, blockSize=7)
|
2019-04-29 09:50:26 +08:00
|
|
|
disparity = stereo.compute(imgL, imgR)
|
2019-04-29 14:10:03 +08:00
|
|
|
plt.title("SGBM")
|
2019-04-29 09:50:26 +08:00
|
|
|
plt.imshow(disparity)
|
|
|
|
|
plt.show()
|
