transfer_pred_disp2depth_v2

tools/eval_utils.py

@@ -119,4 +119,23 @@ def transfer_pred_disp2depth(all_pred_disparity, all_gt_depths, all_masks):
     all_pred_depths = disparity2depth(align_disp_pred)
     return all_pred_depths
 
-    
+    
+
+
+"""
+not gt needed to transfer
+"""
+def transfer_pred_disp2depth_v2(all_pred_disparity, scale, shift):
+    
+
+    ori_shape =all_pred_disparity.shape
+    tmp = all_pred_disparity.squeeze()
+    tmp = tmp * scale + shift
+    align_disp_pred = tmp.reshape(ori_shape)
+    
+    align_disp_pred = np.clip(
+            align_disp_pred, a_min=1e-3, a_max=None
+        )  # avoid 0 disparity
+    all_pred_depths = disparity2depth(align_disp_pred)
+
+    return all_pred_depths