From c1cb5641c73c14668c3dc4201af788a3ef675cfc Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sun, 22 Sep 2024 10:27:01 -0700 Subject: [PATCH] Add weight fore mobilenetv4 small 0.5, change 0.25 -> 0.35 --- timm/models/mobilenetv3.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/timm/models/mobilenetv3.py b/timm/models/mobilenetv3.py index d430a933..d0b6846a 100644 --- a/timm/models/mobilenetv3.py +++ b/timm/models/mobilenetv3.py @@ -1011,6 +1011,13 @@ default_cfgs = generate_default_cfgs({ ), "lcnet_150.untrained": _cfg(), + 'mobilenetv4_conv_small_035.untrained': _cfg( + mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, + test_input_size=(3, 256, 256), test_crop_pct=0.95, interpolation='bicubic'), + 'mobilenetv4_conv_small_050.e3000_r224_in1k': _cfg( + hf_hub_id='timm/', + mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, + test_input_size=(3, 256, 256), test_crop_pct=0.95, interpolation='bicubic'), 'mobilenetv4_conv_small.e2400_r224_in1k': _cfg( hf_hub_id='timm/', test_input_size=(3, 256, 256), test_crop_pct=0.95, interpolation='bicubic'), @@ -1254,20 +1261,19 @@ def lcnet_150(pretrained: bool = False, **kwargs) -> MobileNetV3: @register_model -def mobilenetv4_conv_small_025(pretrained: bool = False, **kwargs) -> MobileNetV3: +def mobilenetv4_conv_small_035(pretrained: bool = False, **kwargs) -> MobileNetV3: """ MobileNet V4 """ - model = _gen_mobilenet_v4('mobilenetv4_conv_small', 0.25, pretrained=pretrained, **kwargs) + model = _gen_mobilenet_v4('mobilenetv4_conv_small_035', 0.35, pretrained=pretrained, **kwargs) return model @register_model def mobilenetv4_conv_small_050(pretrained: bool = False, **kwargs) -> MobileNetV3: """ MobileNet V4 """ - model = _gen_mobilenet_v4('mobilenetv4_conv_small', 0.50, pretrained=pretrained, **kwargs) + model = _gen_mobilenet_v4('mobilenetv4_conv_small_050', 0.50, pretrained=pretrained, **kwargs) return model - @register_model def mobilenetv4_conv_small(pretrained: bool = False, **kwargs) -> MobileNetV3: """ MobileNet V4 """