@@ -453,7 +453,8 @@ def _gen_mobilenet_v3_rw(
453453
454454
455455def _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
457458) -> MobileNetV3 :
458459 """Creates a MobileNet-V3 model.
459460
@@ -537,7 +538,7 @@ def _gen_mobilenet_v3(
537538 ]
538539 se_layer = partial (SqueezeExcite , gate_layer = 'hard_sigmoid' , force_act_layer = nn .ReLU , rd_round_fn = round_channels )
539540 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 ),
541542 num_features = num_features ,
542543 stem_size = 16 ,
543544 fix_stem = channel_multiplier < 0.75 ,
@@ -927,6 +928,9 @@ def _cfg(url: str = '', **kwargs):
927928 origin_url = 'https://github.com/Alibaba-MIIL/ImageNet21K' ,
928929 paper_ids = 'arXiv:2104.10972v4' ,
929930 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+ ),
930934
931935 'mobilenetv3_small_050.lamb_in1k' : _cfg (
932936 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:
10991103 model = _gen_mobilenet_v3 ('mobilenetv3_large_100' , 1.0 , pretrained = pretrained , ** kwargs )
11001104 return model
11011105
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
11021111
11031112@register_model
11041113def mobilenetv3_small_050 (pretrained : bool = False , ** kwargs ) -> MobileNetV3 :
0 commit comments