Merge branch 'baorepo-patch-1'

huggingface · Sep 22, 2024 · 305a360 · 305a360
2 parents 9337463 + c1cb564
commit 305a360
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/timm/models/mobilenetv3.py b/timm/models/mobilenetv3.py
@@ -1011,6 +1011,13 @@ def _cfg(url: str = '', **kwargs):
     ),
     "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'),
@@ -1253,6 +1260,20 @@ def lcnet_150(pretrained: bool = False, **kwargs) -> MobileNetV3:
     return model
 
 
+@register_model
+def mobilenetv4_conv_small_035(pretrained: bool = False, **kwargs) -> MobileNetV3:
+    """ MobileNet V4 """
+    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_050', 0.50, pretrained=pretrained, **kwargs)
+    return model
+
+
 @register_model
 def mobilenetv4_conv_small(pretrained: bool = False, **kwargs) -> MobileNetV3:
     """ MobileNet V4 """