Skip to content

Commit

Permalink
Expose DPT depth models via torch.hub.load() (#238)
Browse files Browse the repository at this point in the history
Add streamlined model versions w/o the mmcv dependency to directly load them via torch.hub.load().
  • Loading branch information
patricklabatut authored Sep 30, 2023
1 parent 82185b1 commit e7df9fc
Show file tree
Hide file tree
Showing 4 changed files with 575 additions and 10 deletions.
2 changes: 1 addition & 1 deletion dinov2/hub/depth/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Loading

0 comments on commit e7df9fc

Please sign in to comment.