Skip to content

Commit 00182fc

Browse files
authored
add tf_mobilenetv3_small_075 func
1 parent 0806ac0 commit 00182fc

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

model/backbone/MobileNetV3.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -621,3 +621,12 @@ def tf_mobilenetv3_large_minimal_100(pretrained=False, **kwargs):
621621
kwargs['pad_type'] = 'same'
622622
model = _gen_mobilenet_v3('tf_mobilenetv3_large_minimal_100', 1.0, pretrained=pretrained, **kwargs)
623623
return model
624+
625+
@register_model
626+
def tf_mobilenetv3_small_075(pretrained=False, **kwargs):
627+
""" MobileNet V3 """
628+
kwargs['bn_eps'] = BN_EPS_TF_DEFAULT
629+
kwargs['pad_type'] = 'same'
630+
model = _gen_mobilenet_v3('tf_mobilenetv3_small_075', 0.75, pretrained=pretrained, **kwargs)
631+
return model
632+

0 commit comments

Comments
 (0)