|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
from dataclasses import dataclass |
|
from typing import Optional |
|
|
|
from .radio_model import Resolution |
|
|
|
|
|
@dataclass |
|
class RadioResource: |
|
url: str |
|
patch_size: int |
|
max_resolution: int |
|
preferred_resolution: Resolution |
|
vitdet_num_windowed: Optional[int] = None |
|
vitdet_num_global: Optional[int] = None |
|
|
|
|
|
RESOURCE_MAP = { |
|
|
|
"radio_v2.5-b": RadioResource( |
|
"https://huggingface.co/nvidia/RADIO/resolve/main/radio-v2.5-b_half.pth.tar?download=true", |
|
patch_size=16, |
|
max_resolution=2048, |
|
preferred_resolution=(768, 768), |
|
vitdet_num_global=4, |
|
), |
|
"radio_v2.5-l": RadioResource( |
|
"https://huggingface.co/nvidia/RADIO/resolve/main/radio-v2.5-l_half.pth.tar?download=true", |
|
patch_size=16, |
|
max_resolution=2048, |
|
preferred_resolution=(768, 768), |
|
vitdet_num_global=4, |
|
), |
|
"radio_v2.5-h": RadioResource( |
|
"https://huggingface.co/nvidia/RADIO/resolve/main/radio_v2.5-h.pth.tar?download=true", |
|
patch_size=16, |
|
max_resolution=2048, |
|
preferred_resolution=(768, 768), |
|
vitdet_num_global=4, |
|
), |
|
"radio_v2.5-h-norm": RadioResource( |
|
"https://huggingface.co/nvidia/RADIO/resolve/main/radio_v2.5-h-norm.pth.tar?download=true", |
|
patch_size=16, |
|
max_resolution=2048, |
|
preferred_resolution=(768, 768), |
|
vitdet_num_global=4, |
|
), |
|
|
|
"radio_v2.1": RadioResource( |
|
"https://huggingface.co/nvidia/RADIO/resolve/main/radio_v2.1_bf16.pth.tar?download=true", |
|
patch_size=16, |
|
max_resolution=2048, |
|
preferred_resolution=Resolution(432, 432), |
|
vitdet_num_windowed=5, |
|
), |
|
"radio_v2": RadioResource( |
|
"https://huggingface.co/nvidia/RADIO/resolve/main/radio_v2.pth.tar?download=true", |
|
patch_size=16, |
|
max_resolution=2048, |
|
preferred_resolution=Resolution(432, 432), |
|
vitdet_num_windowed=5, |
|
), |
|
"radio_v1": RadioResource( |
|
"https://huggingface.co/nvidia/RADIO/resolve/main/radio_v1.pth.tar?download=true", |
|
patch_size=14, |
|
max_resolution=1050, |
|
preferred_resolution=Resolution(378, 378), |
|
), |
|
|
|
"e-radio_v2": RadioResource( |
|
"https://huggingface.co/nvidia/RADIO/resolve/main/eradio_v2.pth.tar?download=true", |
|
patch_size=16, |
|
max_resolution=2048, |
|
preferred_resolution=Resolution(512, 512), |
|
), |
|
|
|
"radio_dinov2-g": RadioResource( |
|
None, |
|
patch_size=14, |
|
max_resolution=2044, |
|
preferred_resolution=Resolution(518, 518), |
|
), |
|
} |
|
|
|
DEFAULT_VERSION = "radio_v2.5-h" |
|
|