diff --git a/doc/img/output.jpg b/doc/img/output.jpg index 6fc8447..c2de955 100644 Binary files a/doc/img/output.jpg and b/doc/img/output.jpg differ diff --git a/test_model.py b/test_model.py index 7e32363..90e3b8a 100644 --- a/test_model.py +++ b/test_model.py @@ -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)