Sandeepa commited on
Commit
1db0f5c
·
1 Parent(s): 39a4453

Update depth_detection.py

Browse files
Files changed (1) hide show
  1. depth_detection.py +2 -2
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
- image = image.crop((pad, pad, image.width - pad, image.height - pad))
19
- return new_img,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, image.width - pad, image.height - pad))
19
+ return new_image,output