add lcnet_050 func

Wuhall · Dec 21, 2022 · 6f2b5d2 · 6f2b5d2
1 parent 1176725
commit 6f2b5d2
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/model/backbone/MobileNetV3.py b/model/backbone/MobileNetV3.py
@@ -670,3 +670,9 @@ def lcnet_035(pretrained=False, **kwargs):
     model = _gen_lcnet('lcnet_035', 0.35, pretrained=pretrained, **kwargs)
     return model
 
+@register_model
+def lcnet_050(pretrained=False, **kwargs):
+    """ PP-LCNet 0.5"""
+    model = _gen_lcnet('lcnet_050', 0.5, pretrained=pretrained, **kwargs)
+    return model
+