Expose linear depth models via PyTorch Hub (#237)
Add streamlined model versions w/o the mmcv dependency to directly load them via torch.hub.load().
Showing
- dinov2/hub/depth/__init__.py 7 additions, 0 deletionsdinov2/hub/depth/__init__.py
- dinov2/hub/depth/decode_heads.py 287 additions, 0 deletionsdinov2/hub/depth/decode_heads.py
- dinov2/hub/depth/encoder_decoder.py 351 additions, 0 deletionsdinov2/hub/depth/encoder_decoder.py
- dinov2/hub/depth/ops.py 28 additions, 0 deletionsdinov2/hub/depth/ops.py
- dinov2/hub/depthers.py 142 additions, 0 deletionsdinov2/hub/depthers.py
- dinov2/hub/utils.py 27 additions, 0 deletionsdinov2/hub/utils.py
- hubconf.py 2 additions, 0 deletionshubconf.py
Loading
Please register or sign in to comment