Sandeepa commited on
Commit
61a5944
·
1 Parent(s): 1db0f5c

Update depth_detection.py

Browse files
Files changed (1) hide show
  1. depth_detection.py +1 -1
depth_detection.py CHANGED
@@ -15,5 +15,5 @@ def depth_detection(image, pad=16):
15
 
16
  output = predicted_depth.squeeze().cpu().numpy() * 1000.0
17
  output = output[pad:-pad, pad:-pad]
18
- new_image = new_img.crop((pad, pad, image.width - pad, image.height - pad))
19
  return new_image,output
 
15
 
16
  output = predicted_depth.squeeze().cpu().numpy() * 1000.0
17
  output = output[pad:-pad, pad:-pad]
18
+ new_image = new_img.crop((pad, pad, new_img.width - pad, new_img.height - pad))
19
  return new_image,output