Spaces:
Running
on
Zero
Running
on
Zero
Update ppd/models/depth_anything_v2/dpt.py
Browse files
ppd/models/depth_anything_v2/dpt.py
CHANGED
|
@@ -188,9 +188,9 @@ class DepthAnythingV2(nn.Module):
|
|
| 188 |
|
| 189 |
# patch_h, patch_w = x.shape[-2] // 14, x.shape[-1] // 14
|
| 190 |
# features = self.pretrained.get_intermediate_layers(x, self.intermediate_layer_idx[self.encoder], return_class_token=True)
|
| 191 |
-
|
| 192 |
|
| 193 |
-
return
|
| 194 |
|
| 195 |
@torch.no_grad()
|
| 196 |
def infer_image(self, raw_image, input_size=518):
|
|
|
|
| 188 |
|
| 189 |
# patch_h, patch_w = x.shape[-2] // 14, x.shape[-1] // 14
|
| 190 |
# features = self.pretrained.get_intermediate_layers(x, self.intermediate_layer_idx[self.encoder], return_class_token=True)
|
| 191 |
+
semantics = self.pretrained.forward_features(x)["x_norm_patchtokens"]
|
| 192 |
|
| 193 |
+
return semantics
|
| 194 |
|
| 195 |
@torch.no_grad()
|
| 196 |
def infer_image(self, raw_image, input_size=518):
|