Spaces:
Running
on
Zero
Running
on
Zero
File size: 702 Bytes
1834bc0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
import timm
from backbones.dino_v2 import build_dinov2_small, build_dinov2_base, build_dinov2_large
from backbones.dino_v3 import build_dinov3_small, build_dinov3_base, build_dinov3_large
"""
Model registry for backbones
"""
_BACKBONES = {
"dinov2_small": build_dinov2_small,
"dinov2_base": build_dinov2_base,
"dinov2_large": build_dinov2_large,
"dinov3_small": build_dinov3_small,
"dinov3_base": build_dinov3_base,
"dinov3_large": build_dinov3_large,
}
def get_backbone(name: str, **kwargs):
if name not in _BACKBONES:
raise ValueError(f"Unknown backbone '{name}'. Available: {list(_BACKBONES)}")
return _BACKBONES[name](**kwargs)
|