JingShiang Yang commited on
Commit
baf0224
·
1 Parent(s): 2454e32
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def analyze_depth(image):
16
  if image is None:
17
  return None, None, None, None, None
18
 
19
- prediction = model.inference([image], process_res_method='lower_bound_resize')
20
 
21
  depth_np = prediction.depth[0].cpu().numpy() if torch.is_tensor(prediction.depth[0]) else prediction.depth[0]
22
  extrinsics_np = prediction.extrinsics[0].cpu().numpy() if torch.is_tensor(prediction.extrinsics[0]) else prediction.extrinsics[0]
 
16
  if image is None:
17
  return None, None, None, None, None
18
 
19
+ prediction = model.inference([image], process_res_method='lower_bound_resize', process_res=1008)
20
 
21
  depth_np = prediction.depth[0].cpu().numpy() if torch.is_tensor(prediction.depth[0]) else prediction.depth[0]
22
  extrinsics_np = prediction.extrinsics[0].cpu().numpy() if torch.is_tensor(prediction.extrinsics[0]) else prediction.extrinsics[0]