Cpt.Captain/CREStereo-pytorch-nxt
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Update output image

Browse Source
This commit is contained in:
Ibai 2022-04-08 11:27:29 +09:00
parent 1bdbb27240
commit c196d6caa3
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
doc/img/output.jpg
View File

Binary file not shown.
Side by Side Swipe Overlay

Before

Width:  |  Height:  |  Size: 20 KiB

After

Width:  |  Height:  |  Size: 119 KiB

6
test_model.py
View File

@ -54,9 +54,11 @@ if __name__ == '__main__':
disp_vis = (disp - disp.min()) / (disp.max() - disp.min()) * 255.0
disp_vis = disp_vis.astype("uint8")
disp_vis = cv2.applyColorMap(disp_vis, cv2.COLORMAP_INFERNO)
disp_vis = cv2.resize(disp_vis, left_img.shape[1::-1])
cv2.imshow("output", disp_vis)
cv2.imwrite("output.jpg", disp_vis)
combined_img = np.hstack((left_img, disp_vis))
cv2.imshow("output", combined_img)
cv2.imwrite("output.jpg", combined_img)
cv2.waitKey(0)