Skip to content

Commit d6b8816

Browse files
committed
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.
1 parent ef570e2 commit d6b8816

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

timm/models/mobilenetv3.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1253,6 +1253,21 @@ def lcnet_150(pretrained: bool = False, **kwargs) -> MobileNetV3:
12531253
return model
12541254

12551255

1256+
@register_model
1257+
def mobilenetv4_conv_small_025(pretrained: bool = False, **kwargs) -> MobileNetV3:
1258+
""" MobileNet V4 """
1259+
model = _gen_mobilenet_v4('mobilenetv4_conv_small', 0.25, pretrained=pretrained, **kwargs)
1260+
return model
1261+
1262+
1263+
@register_model
1264+
def mobilenetv4_conv_small_050(pretrained: bool = False, **kwargs) -> MobileNetV3:
1265+
""" MobileNet V4 """
1266+
model = _gen_mobilenet_v4('mobilenetv4_conv_small', 0.50, pretrained=pretrained, **kwargs)
1267+
return model
1268+
1269+
1270+
12561271
@register_model
12571272
def mobilenetv4_conv_small(pretrained: bool = False, **kwargs) -> MobileNetV3:
12581273
""" MobileNet V4 """

0 commit comments

Comments
 (0)