Update depth_anything_v2/dpt.py
Browse files- depth_anything_v2/dpt.py +1 -1
depth_anything_v2/dpt.py
CHANGED
|
@@ -191,7 +191,7 @@ class DepthAnythingV2(nn.Module):
|
|
| 191 |
|
| 192 |
depth = F.interpolate(depth[:, None], (h, w), mode="bilinear", align_corners=True)[0, 0]
|
| 193 |
|
| 194 |
-
return depth
|
| 195 |
|
| 196 |
def image2tensor(self, raw_image, input_size=518):
|
| 197 |
transform = Compose([
|
|
|
|
| 191 |
|
| 192 |
depth = F.interpolate(depth[:, None], (h, w), mode="bilinear", align_corners=True)[0, 0]
|
| 193 |
|
| 194 |
+
return depth.cpu().numpy()
|
| 195 |
|
| 196 |
def image2tensor(self, raw_image, input_size=518):
|
| 197 |
transform = Compose([
|