add fbnetv3_b func

Wuhall · Dec 21, 2022 · cd02485 · cd02485
1 parent 21af92d
commit cd02485
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/model/backbone/MobileNetV3.py b/model/backbone/MobileNetV3.py
@@ -645,3 +645,9 @@ def tf_mobilenetv3_small_minimal_100(pretrained=False, **kwargs):
     kwargs['pad_type'] = 'same'
     model = _gen_mobilenet_v3('tf_mobilenetv3_small_minimal_100', 1.0, pretrained=pretrained, **kwargs)
     return model
+
+@register_model
+def fbnetv3_b(pretrained=False, **kwargs):
+    """ FBNetV3-B """
+    model = _gen_fbnetv3('fbnetv3_b', pretrained=pretrained, **kwargs)
+    return model