add tf_mobilenetv3_large_075 func

Wuhall · Dec 21, 2022 · 0d31669 · 0d31669
1 parent 550f0a2
commit 0d31669
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/model/backbone/MobileNetV3.py b/model/backbone/MobileNetV3.py
@@ -597,3 +597,11 @@ def mobilenetv3_rw(pretrained=False, **kwargs):
         kwargs['bn_eps'] = BN_EPS_TF_DEFAULT
     model = _gen_mobilenet_v3_rw('mobilenetv3_rw', 1.0, pretrained=pretrained, **kwargs)
     return model
+
+@register_model
+def tf_mobilenetv3_large_075(pretrained=False, **kwargs):
+    """ MobileNet V3 """
+    kwargs['bn_eps'] = BN_EPS_TF_DEFAULT
+    kwargs['pad_type'] = 'same'
+    model = _gen_mobilenet_v3('tf_mobilenetv3_large_075', 0.75, pretrained=pretrained, **kwargs)
+    return model