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
Download Patch File Download Diff File Expand all files Collapse all files
doc/img/output.jpg
BIN
View File
Binary file not shown.
Side by Side Swipe Overlay

Before

Width:  |  Height:  |  Size: 20 KiB

After

Width:  |  Height:  |  Size: 119 KiB

test_model.py
+4 -2
View File
@@ -54,9 +54,11 @@ if __name__ == '__main__':
disp_vis = (disp - disp.min()) / (disp.max() - disp.min()) * 255.0 disp_vis = (disp - disp.min()) / (disp.max() - disp.min()) * 255.0
disp_vis = disp_vis.astype("uint8") disp_vis = disp_vis.astype("uint8")
disp_vis = cv2.applyColorMap(disp_vis, cv2.COLORMAP_INFERNO) 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) combined_img = np.hstack((left_img, disp_vis))
cv2.imwrite("output.jpg", disp_vis) cv2.imshow("output", combined_img)
cv2.imwrite("output.jpg", combined_img)
cv2.waitKey(0) cv2.waitKey(0)