mirrors
/
dinov2
mirror of https://github.com/facebookresearch/dinov2.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
dinov2/hubconf.py

16 lines
858 B
Python
Raw Normal View History

Initial commit
2023-04-18 01:55:51 +08:00
# Copyright (c) Meta Platforms, Inc. and affiliates.
#
Update license everywhere (#182) Update code and models license from CC-BY-NC to Apache 2.0 in headers and other files.
2023-08-31 20:41:52 +08:00
# 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.
Initial commit
2023-04-18 01:55:51 +08:00
Rework Pytorch Hub support code (#202) Rework support code for torch.hub.load() to allow reusing shared functions and eventually expose more models.
2023-09-27 23:06:03 +08:00
from dinov2.hub.backbones import dinov2_vitb14, dinov2_vitg14, dinov2_vitl14, dinov2_vits14
Add new backbones trained with registers (#282) Add new backbones (and matching linear classification heads) trained with 4 registers following [Vision Transformers Need Registers](https://arxiv.org/abs/2309.16588).
2023-10-27 21:15:10 +08:00
from dinov2.hub.backbones import dinov2_vitb14_reg, dinov2_vitg14_reg, dinov2_vitl14_reg, dinov2_vits14_reg
Rework Pytorch Hub support code (#202) Rework support code for torch.hub.load() to allow reusing shared functions and eventually expose more models.
2023-09-27 23:06:03 +08:00
from dinov2.hub.classifiers import dinov2_vitb14_lc, dinov2_vitg14_lc, dinov2_vitl14_lc, dinov2_vits14_lc
Add new backbones trained with registers (#282) Add new backbones (and matching linear classification heads) trained with 4 registers following [Vision Transformers Need Registers](https://arxiv.org/abs/2309.16588).
2023-10-27 21:15:10 +08:00
from dinov2.hub.classifiers import dinov2_vitb14_reg_lc, dinov2_vitg14_reg_lc, dinov2_vitl14_reg_lc, dinov2_vits14_reg_lc
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().
2023-10-01 02:06:05 +08:00
from dinov2.hub.depthers import dinov2_vitb14_ld, dinov2_vitg14_ld, dinov2_vitl14_ld, dinov2_vits14_ld
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().
2023-10-01 02:12:02 +08:00
from dinov2.hub.depthers import dinov2_vitb14_dd, dinov2_vitg14_dd, dinov2_vitl14_dd, dinov2_vits14_dd
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().
2023-10-01 02:06:05 +08:00
Initial commit
2023-04-18 01:55:51 +08:00
dependencies = ["torch"]