Expose DPT depth models via torch.hub.load() (#238)

Add streamlined model versions w/o the mmcv dependency to directly load them via torch.hub.load().
facebookresearch · Sep 30, 2023 · e7df9fc · e7df9fc
1 parent 82185b1
commit e7df9fc
Show file tree

Hide file tree

Showing 4 changed files with 575 additions and 10 deletions.
diff --git a/dinov2/hub/depth/__init__.py b/dinov2/hub/depth/__init__.py
@@ -3,5 +3,5 @@
 # This source code is licensed under the Apache License, Version 2.0
 # found in the LICENSE file in the root directory of this source tree.
 
-from .decode_heads import BNHead
+from .decode_heads import BNHead, DPTHead
 from .encoder_decoder import DepthEncoderDecoder