From 146c2fbe3481248f044d1545aa765b032f35850d Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 16 Aug 2024 12:10:00 -0700 Subject: [PATCH] Add resnet50d and efficientnet_b1 ra4 (mnv4) hparam weights --- timm/models/efficientnet.py | 8 ++++++-- timm/models/resnet.py | 5 +++++ 2 files changed, 11 insertions(+), 2 deletions(-) diff --git a/timm/models/efficientnet.py b/timm/models/efficientnet.py index 3aee0342..09d6c66c 100644 --- a/timm/models/efficientnet.py +++ b/timm/models/efficientnet.py @@ -1290,8 +1290,12 @@ default_cfgs = generate_default_cfgs({ 'efficientnet_b0.ra4_e3600_r224_in1k': _cfg( hf_hub_id='timm/', mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, - crop_pct=0.9, test_input_size=(3, 256, 256), test_crop_pct=1.0 - ), + crop_pct=0.9, test_input_size=(3, 256, 256), test_crop_pct=1.0), + 'efficientnet_b1.ra4_e3600_r240_in1k': _cfg( + hf_hub_id='timm/', + mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, + input_size=(3, 240, 240), crop_pct=0.9, + test_input_size=(3, 288, 288), test_crop_pct=1.0), 'efficientnet_b1.ft_in1k': _cfg( url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/efficientnet_b1-533bc792.pth', hf_hub_id='timm/', diff --git a/timm/models/resnet.py b/timm/models/resnet.py index 1d60deca..a80954cc 100644 --- a/timm/models/resnet.py +++ b/timm/models/resnet.py @@ -783,6 +783,11 @@ default_cfgs = generate_default_cfgs({ hf_hub_id='timm/', url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/resnet50d_ra2-464e36ba.pth', first_conv='conv1.0'), + 'resnet50d.ra4_e3600_r224_in1k': _rcfg( + hf_hub_id='timm/', + mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5), + crop_pct=0.95, test_input_size=(3, 288, 288), test_crop_pct=1.0, + first_conv='conv1.0'), 'resnet50d.a1_in1k': _rcfg( hf_hub_id='timm/', url='https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-rsb-weights/resnet50d_a1_0-e20cff14.pth',