Spaces:
Runtime error
Runtime error
Update mono/utils/do_test.py
Browse files- mono/utils/do_test.py +2 -1
mono/utils/do_test.py
CHANGED
@@ -152,7 +152,8 @@ def get_prediction(
|
|
152 |
input=input,
|
153 |
cam_model=cam_model,
|
154 |
)
|
155 |
-
pred_depth, confidence, output_dict = model.module.inference(data)
|
|
|
156 |
pred_depth = pred_depth
|
157 |
pred_depth = pred_depth.squeeze()
|
158 |
pred_depth = pred_depth[pad_info[0] : pred_depth.shape[0] - pad_info[1], pad_info[2] : pred_depth.shape[1] - pad_info[3]]
|
|
|
152 |
input=input,
|
153 |
cam_model=cam_model,
|
154 |
)
|
155 |
+
#pred_depth, confidence, output_dict = model.module.inference(data)
|
156 |
+
pred_depth, confidence, output_dict = model.inference(data)
|
157 |
pred_depth = pred_depth
|
158 |
pred_depth = pred_depth.squeeze()
|
159 |
pred_depth = pred_depth[pad_info[0] : pred_depth.shape[0] - pad_info[1], pad_info[2] : pred_depth.shape[1] - pad_info[3]]
|