mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

MobilenetV4: add two more lightweight models

Browse Source 
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.
This commit is contained in:
alias pillar1989 2024-09-05 10:27:48 +08:00 committed by Baozhu Zuo
parent ef570e266d
commit d6b8816eda
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
timm/models/mobilenetv3.py
View File

@ -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 """