From 800405d9416a8d7b3917a815ce24c780828969bf Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 8 Jul 2024 13:50:05 -0700 Subject: [PATCH] Add conv_large mobilenetv3 aa/blur model defs --- timm/models/mobilenetv3.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/timm/models/mobilenetv3.py b/timm/models/mobilenetv3.py index 8aa1acb0..ec820aec 100644 --- a/timm/models/mobilenetv3.py +++ b/timm/models/mobilenetv3.py @@ -1045,6 +1045,14 @@ default_cfgs = generate_default_cfgs({ 'mobilenetv4_conv_blur_medium.e500_r224_in1k': _cfg( hf_hub_id='timm/', crop_pct=0.95, test_input_size=(3, 256, 256), test_crop_pct=1.0, interpolation='bicubic'), + 'mobilenetv4_conv_aa_large.e600_r384_in1k': _cfg( + # hf_hub_id='timm/', + input_size=(3, 384, 384), pool_size=(12, 12), + crop_pct=0.95, test_input_size=(3, 448, 448), test_crop_pct=1.0, interpolation='bicubic'), + 'mobilenetv4_conv_blur_large.e600_r384_in1k': _cfg( + # hf_hub_id='timm/', + input_size=(3, 384, 384), pool_size=(12, 12), + crop_pct=0.95, test_input_size=(3, 448, 448), test_crop_pct=1.0, interpolation='bicubic'), 'mobilenetv4_hybrid_medium_075.untrained': _cfg( # hf_hub_id='timm/', crop_pct=0.95, interpolation='bicubic'), @@ -1256,6 +1264,20 @@ def mobilenetv4_conv_blur_medium(pretrained: bool = False, **kwargs) -> MobileNe return model +@register_model +def mobilenetv4_conv_aa_large(pretrained: bool = False, **kwargs) -> MobileNetV3: + """ MobileNet V4 w/ AvgPool AA """ + model = _gen_mobilenet_v4('mobilenetv4_conv_aa_large', 1.0, pretrained=pretrained, aa_layer='avg', **kwargs) + return model + + +@register_model +def mobilenetv4_conv_blur_large(pretrained: bool = False, **kwargs) -> MobileNetV3: + """ MobileNet V4 Conv w/ Blur AA """ + model = _gen_mobilenet_v4('mobilenetv4_conv_blur_large', 1.0, pretrained=pretrained, aa_layer='blurpc', **kwargs) + return model + + @register_model def mobilenetv4_hybrid_medium_075(pretrained: bool = False, **kwargs) -> MobileNetV3: """ MobileNet V4 Hybrid """