Skip to content

Commit ebbe530

Browse files
committed
Add MobileNetV3 RA4 (mnv4 recipe) weights
1 parent b3f3a00 commit ebbe530

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

timm/models/mobilenetv3.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -916,6 +916,10 @@ def _cfg(url: str = '', **kwargs):
916916
interpolation='bicubic',
917917
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mobilenetv3_large_100_ra-f55367f5.pth',
918918
hf_hub_id='timm/'),
919+
'mobilenetv3_large_100.ra4_e3600_r224_in1k': _cfg(
920+
hf_hub_id='timm/',
921+
interpolation='bicubic', mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD,
922+
crop_pct=0.95, test_input_size=(3, 256, 256), test_crop_pct=1.0),
919923
'mobilenetv3_large_100.miil_in21k_ft_in1k': _cfg(
920924
interpolation='bilinear', mean=(0., 0., 0.), std=(1., 1., 1.),
921925
origin_url='https://github.com/Alibaba-MIIL/ImageNet21K',
@@ -928,9 +932,10 @@ def _cfg(url: str = '', **kwargs):
928932
origin_url='https://github.com/Alibaba-MIIL/ImageNet21K',
929933
paper_ids='arXiv:2104.10972v4',
930934
interpolation='bilinear', mean=(0., 0., 0.), std=(1., 1., 1.), num_classes=11221),
931-
'mobilenetv3_large_150d.untrained': _cfg(
932-
#hf_hub_id='timm/',
933-
),
935+
'mobilenetv3_large_150d.ra4_e3600_r256_in1k': _cfg(
936+
hf_hub_id='timm/',
937+
mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD,
938+
input_size=(3, 256, 256), crop_pct=0.95, pool_size=(8, 8), test_input_size=(3, 320, 320), test_crop_pct=1.0),
934939

935940
'mobilenetv3_small_050.lamb_in1k': _cfg(
936941
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mobilenetv3_small_050_lambc-4b7bbe87.pth',

0 commit comments

Comments
 (0)