Spaces:
Runtime error
Runtime error
Update depth_detection.py
Browse files- 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 |
-
|
19 |
-
return
|
|
|
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
|