Spaces:

depth-anything
/

Depth-Anything-V2

Running on Zero

LiheYoung commited on Jun 15

Commit

8686747

•

1 Parent(s): 0d402f2

Update depth_anything_v2/dpt.py

Files changed (1) hide show

depth_anything_v2/dpt.py CHANGED Viewed

@@ -191,7 +191,7 @@ class DepthAnythingV2(nn.Module):
         depth = F.interpolate(depth[:, None], (h, w), mode="bilinear", align_corners=True)[0, 0]
-        return depth
     def image2tensor(self, raw_image, input_size=518):
         transform = Compose([

         depth = F.interpolate(depth[:, None], (h, w), mode="bilinear", align_corners=True)[0, 0]
+        return depth.cpu().numpy()
     def image2tensor(self, raw_image, input_size=518):
         transform = Compose([