From d6b8816eda77996993869dc7f0c55f0b7753d9de Mon Sep 17 00:00:00 2001 From: alias pillar1989 Date: Thu, 5 Sep 2024 10:27:48 +0800 Subject: [PATCH] MobilenetV4: add two more lightweight models Mobilenetv4 is very fast and ideal for embedded devices. However, for many low-cost, low-power embedded MCU devices, smaller models are required. Hopefully this PR will merge. --- timm/models/mobilenetv3.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/timm/models/mobilenetv3.py b/timm/models/mobilenetv3.py index 0aa0eebc..d430a933 100644 --- a/timm/models/mobilenetv3.py +++ b/timm/models/mobilenetv3.py @@ -1253,6 +1253,21 @@ def lcnet_150(pretrained: bool = False, **kwargs) -> MobileNetV3: return model +@register_model +def mobilenetv4_conv_small_025(pretrained: bool = False, **kwargs) -> MobileNetV3: + """ MobileNet V4 """ + model = _gen_mobilenet_v4('mobilenetv4_conv_small', 0.25, 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) + return model + + + @register_model def mobilenetv4_conv_small(pretrained: bool = False, **kwargs) -> MobileNetV3: """ MobileNet V4 """