JUGGHM commited on
Commit
cc17fb8
1 Parent(s): 71365ca

Update mono/utils/do_test.py

Browse files
Files changed (1) hide show
  1. 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]]