From cd02485941d45968b04bdc4810f40ce6220ce6ae Mon Sep 17 00:00:00 2001
From: wmkai <39148461+wmkai@users.noreply.github.com>
Date: Wed, 21 Dec 2022 17:11:23 +0800
Subject: [PATCH] add fbnetv3_b func

---
 model/backbone/MobileNetV3.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/model/backbone/MobileNetV3.py b/model/backbone/MobileNetV3.py
index 74ce8f5..0cab8fe 100644
--- 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