You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30 lines
452 B
Python
30 lines
452 B
Python
from .dinov2 import (
|
|
DINOv2_base,
|
|
DINOv2_base_reg,
|
|
DINOv2_large,
|
|
DINOv2_large_reg,
|
|
DINOv2_small,
|
|
DINOv2_small_reg,
|
|
)
|
|
from .vit import (
|
|
ViT,
|
|
ViT_base,
|
|
ViT_large,
|
|
ViT_small,
|
|
ViT_tiny,
|
|
)
|
|
|
|
__all__ = [
|
|
"DINOv2_base",
|
|
"DINOv2_base_reg",
|
|
"DINOv2_large",
|
|
"DINOv2_large_reg",
|
|
"DINOv2_small",
|
|
"DINOv2_small_reg",
|
|
"ViT",
|
|
"ViT_base",
|
|
"ViT_large",
|
|
"ViT_small",
|
|
"ViT_tiny",
|
|
]
|