You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
21 lines
791 B
21 lines
791 B
import numpy as np |
|
import cv2 as cv |
|
|
|
left_img = cv.imread(cv.samples.findFile("aloeL.jpg"), cv.IMREAD_COLOR) |
|
right_img = cv.imread(cv.samples.findFile("aloeR.jpg"), cv.IMREAD_COLOR) |
|
|
|
frame_size = left_img.shape[0:2]; |
|
|
|
stereo = cv.stereo.QuasiDenseStereo_create(frame_size[::-1]) |
|
stereo.process(left_img, right_img) |
|
disp = stereo.getDisparity() |
|
cv.imshow("disparity", disp) |
|
cv.waitKey() |
|
dense_matches = stereo.getDenseMatches() |
|
try: |
|
with open("dense.txt", "wt") as f: |
|
# if you want all matches use for idx in len(dense_matches): It can be a big file |
|
for idx in range(0, min(10, len(dense_matches))): |
|
nb = f.write(str(dense_matches[idx].p0) + "\t" + str(dense_matches[idx].p1) + "\t" + str(dense_matches[idx].corr) + "\n") |
|
except: |
|
print("Cannot open file")
|
|
|