@@ -453,7 +453,8 @@ def _gen_mobilenet_v3_rw(
453
453
454
454
455
455
def _gen_mobilenet_v3 (
456
- variant : str , channel_multiplier : float = 1.0 , group_size = None , pretrained : bool = False , ** kwargs
456
+ variant : str , channel_multiplier : float = 1.0 , depth_multiplier : float = 1.0 ,
457
+ group_size = None , pretrained : bool = False , ** kwargs
457
458
) -> MobileNetV3 :
458
459
"""Creates a MobileNet-V3 model.
459
460
@@ -537,7 +538,7 @@ def _gen_mobilenet_v3(
537
538
]
538
539
se_layer = partial (SqueezeExcite , gate_layer = 'hard_sigmoid' , force_act_layer = nn .ReLU , rd_round_fn = round_channels )
539
540
model_kwargs = dict (
540
- block_args = decode_arch_def (arch_def , group_size = group_size ),
541
+ block_args = decode_arch_def (arch_def , depth_multiplier = depth_multiplier , group_size = group_size ),
541
542
num_features = num_features ,
542
543
stem_size = 16 ,
543
544
fix_stem = channel_multiplier < 0.75 ,
@@ -927,6 +928,9 @@ def _cfg(url: str = '', **kwargs):
927
928
origin_url = 'https://github.com/Alibaba-MIIL/ImageNet21K' ,
928
929
paper_ids = 'arXiv:2104.10972v4' ,
929
930
interpolation = 'bilinear' , mean = (0. , 0. , 0. ), std = (1. , 1. , 1. ), num_classes = 11221 ),
931
+ 'mobilenetv3_large_150d.untrained' : _cfg (
932
+ #hf_hub_id='timm/',
933
+ ),
930
934
931
935
'mobilenetv3_small_050.lamb_in1k' : _cfg (
932
936
url = 'https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mobilenetv3_small_050_lambc-4b7bbe87.pth' ,
@@ -1099,6 +1103,11 @@ def mobilenetv3_large_100(pretrained: bool = False, **kwargs) -> MobileNetV3:
1099
1103
model = _gen_mobilenet_v3 ('mobilenetv3_large_100' , 1.0 , pretrained = pretrained , ** kwargs )
1100
1104
return model
1101
1105
1106
+ @register_model
1107
+ def mobilenetv3_large_150d (pretrained : bool = False , ** kwargs ) -> MobileNetV3 :
1108
+ """ MobileNet V3 """
1109
+ model = _gen_mobilenet_v3 ('mobilenetv3_large_150d' , 1.5 , depth_multiplier = 1.2 , pretrained = pretrained , ** kwargs )
1110
+ return model
1102
1111
1103
1112
@register_model
1104
1113
def mobilenetv3_small_050 (pretrained : bool = False , ** kwargs ) -> MobileNetV3 :
0 commit comments