gangweix commited on
Commit
1f69cfa
·
verified ·
1 Parent(s): 5ee6fca

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
- latent = self.pretrained.forward_features(x)["x_norm_patchtokens"]
192
 
193
- return latent
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):