| """Architecture configurations for inference (input sizes only).""" | |
| ARCHITECTURES = { | |
| "vit_tiny_patch16_224.augreg_in21k_ft_in1k": {"input_size": 224}, | |
| "vit_base_patch16_224.augreg_in21k_ft_in1k": {"input_size": 224}, | |
| "efficientvit_b0.r224_in1k": {"input_size": 224}, | |
| "efficientvit_b3.r224_in1k": {"input_size": 224}, | |
| "convnextv2_atto.fcmae_ft_in1k": {"input_size": 224}, | |
| "convnextv2_base.fcmae_ft_in22k_in1k": {"input_size": 224}, | |
| "efficientnetv2_rw_t.ra2_in1k": {"input_size": 224}, | |
| "efficientnetv2_rw_m.agc_in1k": {"input_size": 320}, | |
| "mambaout_tiny.in1k": {"input_size": 224}, | |
| "mambaout_base.in1k": {"input_size": 224}, | |
| "focalnet_tiny_lrf.ms_in1k": {"input_size": 224}, | |
| "focalnet_base_lrf.ms_in1k": {"input_size": 224}, | |
| "edgenext_xx_small.in1k": {"input_size": 256}, | |
| "edgenext_base.in21k_ft_in1k": {"input_size": 256}, | |
| "swin_tiny_patch4_window7_224": {"input_size": 224}, | |
| "swin_base_patch4_window7_224": {"input_size": 224}, | |
| } | |
| def get_default_input_size(architecture_name): | |
| """Get default input size for an architecture.""" | |
| return ARCHITECTURES.get(architecture_name, {}).get("input_size", 224) | |