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]]
|