Skip to content

Commit e6263bf

Browse files
committed
Add resnet and resnet-v2 18/34 weights trained with mnv4 small based recipe
1 parent f5b58e3 commit e6263bf

File tree

2 files changed

+20
-8
lines changed

2 files changed

+20
-8
lines changed

timm/models/resnet.py

+6
Original file line numberDiff line numberDiff line change
@@ -711,6 +711,9 @@ def _gcfg(url='', **kwargs):
711711
hf_hub_id='timm/',
712712
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/resnet18d_ra2-48a79e06.pth',
713713
first_conv='conv1.0'),
714+
'resnet18d.ra4_e3600_r224_in1k': _rcfg(
715+
hf_hub_id='timm/',
716+
mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5), crop_pct=0.9),
714717
'resnet34.a1_in1k': _rcfg(
715718
hf_hub_id='timm/',
716719
url='https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-rsb-weights/resnet34_a1_0-46f8f793.pth'),
@@ -724,6 +727,9 @@ def _gcfg(url='', **kwargs):
724727
'resnet34.bt_in1k': _ttcfg(
725728
hf_hub_id='timm/',
726729
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/resnet34-43635321.pth'),
730+
'resnet34.ra4_e3600_r224_in1k': _rcfg(
731+
hf_hub_id='timm/',
732+
mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5), crop_pct=0.9),
727733
'resnet34d.ra2_in1k': _ttcfg(
728734
hf_hub_id='timm/',
729735
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/resnet34d_ra2-f8dcfcaf.pth',

timm/models/resnetv2.py

+14-8
Original file line numberDiff line numberDiff line change
@@ -696,14 +696,20 @@ def _cfg(url='', **kwargs):
696696
hf_hub_id='timm/',
697697
num_classes=21843, custom_load=True),
698698

699-
'resnetv2_18.untrained': _cfg(
700-
interpolation='bicubic', crop_pct=0.95),
701-
'resnetv2_18d.untrained': _cfg(
702-
interpolation='bicubic', crop_pct=0.95, first_conv='stem.conv1'),
703-
'resnetv2_34.untrained': _cfg(
704-
interpolation='bicubic', crop_pct=0.95),
705-
'resnetv2_34d.untrained': _cfg(
706-
interpolation='bicubic', crop_pct=0.95, first_conv='stem.conv1'),
699+
'resnetv2_18.ra4_e3600_r224_in1k': _cfg(
700+
hf_hub_id='timm/',
701+
interpolation='bicubic', crop_pct=0.9, test_input_size=(3, 288, 288), test_crop_pct=1.0),
702+
'resnetv2_18d.ra4_e3600_r224_in1k': _cfg(
703+
hf_hub_id='timm/',
704+
interpolation='bicubic', crop_pct=0.9, test_input_size=(3, 288, 288), test_crop_pct=1.0,
705+
first_conv='stem.conv1'),
706+
'resnetv2_34.ra4_e3600_r224_in1k': _cfg(
707+
hf_hub_id='timm/',
708+
interpolation='bicubic', crop_pct=0.9, test_input_size=(3, 288, 288), test_crop_pct=1.0),
709+
'resnetv2_34d.ra4_e3600_r224_in1k': _cfg(
710+
hf_hub_id='timm/',
711+
interpolation='bicubic', crop_pct=0.9, test_input_size=(3, 288, 288), test_crop_pct=1.0,
712+
first_conv='stem.conv1'),
707713
'resnetv2_50.a1h_in1k': _cfg(
708714
hf_hub_id='timm/',
709715
interpolation='bicubic', crop_pct=0.95, test_input_size=(3, 288, 288), test_crop_pct=1.0),

0 commit comments

Comments
 (0)