Spaces:

Sandeepa
/

MeshReconstruction3D

Runtime error

Sandeepa commited on May 7, 2023

Commit

39a4453

1 Parent(s): 3078f9c

Update depth_detection.py

Files changed (1) hide show

depth_detection.py CHANGED Viewed

@@ -15,4 +15,5 @@ def depth_detection(image, pad=16):
     output = predicted_depth.squeeze().cpu().numpy() * 1000.0
     output = output[pad:-pad, pad:-pad]
     return new_img,output

     output = predicted_depth.squeeze().cpu().numpy() * 1000.0
     output = output[pad:-pad, pad:-pad]
+    image = image.crop((pad, pad, image.width - pad, image.height - pad))
     return new_img,output