diff --git a/hubconf.py b/hubconf.py index 848d590a0fd750f731bb4e78c6c1c569153bde06..a1c79b689f7194b7220863fcef5fcff9c493391f 100644 --- a/hubconf.py +++ b/hubconf.py @@ -484,8 +484,6 @@ def MobileNetV2_x2_0(**kwargs): return model - - def MobileNetV3_large_x0_35(**kwargs): '''MobileNetV3_large_x0_35 ''' @@ -514,7 +512,6 @@ def MobileNetV3_large_x0_5(**kwargs): return model - def MobileNetV3_large_x0_75(**kwargs): '''MobileNetV3_large_x0_75 ''' @@ -543,7 +540,6 @@ def MobileNetV3_large_x1_0(**kwargs): return model - def MobileNetV3_large_x1_25(**kwargs): '''MobileNetV3_large_x1_25 ''' @@ -558,7 +554,6 @@ def MobileNetV3_large_x1_25(**kwargs): return model - def MobileNetV3_small_x0_35(**kwargs): '''MobileNetV3_small_x0_35 ''' @@ -573,7 +568,6 @@ def MobileNetV3_small_x0_35(**kwargs): return model - def MobileNetV3_small_x0_5(**kwargs): '''MobileNetV3_small_x0_5 ''' @@ -588,7 +582,6 @@ def MobileNetV3_small_x0_5(**kwargs): return model - def MobileNetV3_small_x0_75(**kwargs): '''MobileNetV3_small_x0_75 ''' @@ -603,7 +596,6 @@ def MobileNetV3_small_x0_75(**kwargs): return model - def MobileNetV3_small_x1_0(**kwargs): '''MobileNetV3_small_x1_0 ''' @@ -618,7 +610,6 @@ def MobileNetV3_small_x1_0(**kwargs): return model - def MobileNetV3_small_x1_25(**kwargs): '''MobileNetV3_small_x1_25 ''' @@ -633,3 +624,86 @@ def MobileNetV3_small_x1_25(**kwargs): return model + +def ResNeXt101_32x4d(**kwargs): + '''ResNeXt101_32x4d + ''' + pretrained = kwargs.pop('pretrained', False) + + model = _resnext.ResNeXt101_32x4d(**kwargs) + if pretrained: + assert 'ResNeXt101_32x4d' in _checkpoints, 'Not provide `ResNeXt101_32x4d` pretrained model.' + path = paddle.utils.download.get_weights_path_from_url(_checkpoints['ResNeXt101_32x4d']) + model.set_state_dict(paddle.load(path)) + + return model + + +def ResNeXt101_64x4d(**kwargs): + '''ResNeXt101_64x4d + ''' + pretrained = kwargs.pop('pretrained', False) + + model = _resnext.ResNeXt101_64x4d(**kwargs) + if pretrained: + assert 'ResNeXt101_64x4d' in _checkpoints, 'Not provide `ResNeXt101_64x4d` pretrained model.' + path = paddle.utils.download.get_weights_path_from_url(_checkpoints['ResNeXt101_64x4d']) + model.set_state_dict(paddle.load(path)) + + return model + + +def ResNeXt152_32x4d(**kwargs): + '''ResNeXt152_32x4d + ''' + pretrained = kwargs.pop('pretrained', False) + + model = _resnext.ResNeXt152_32x4d(**kwargs) + if pretrained: + assert 'ResNeXt152_32x4d' in _checkpoints, 'Not provide `ResNeXt152_32x4d` pretrained model.' + path = paddle.utils.download.get_weights_path_from_url(_checkpoints['ResNeXt152_32x4d']) + model.set_state_dict(paddle.load(path)) + + return model + + +def ResNeXt152_64x4d(**kwargs): + '''ResNeXt152_64x4d + ''' + pretrained = kwargs.pop('pretrained', False) + + model = _resnext.ResNeXt152_64x4d(**kwargs) + if pretrained: + assert 'ResNeXt152_64x4d' in _checkpoints, 'Not provide `ResNeXt152_64x4d` pretrained model.' + path = paddle.utils.download.get_weights_path_from_url(_checkpoints['ResNeXt152_64x4d']) + model.set_state_dict(paddle.load(path)) + + return model + + +def ResNeXt50_32x4d(**kwargs): + '''ResNeXt50_32x4d + ''' + pretrained = kwargs.pop('pretrained', False) + + model = _resnext.ResNeXt50_32x4d(**kwargs) + if pretrained: + assert 'ResNeXt50_32x4d' in _checkpoints, 'Not provide `ResNeXt50_32x4d` pretrained model.' + path = paddle.utils.download.get_weights_path_from_url(_checkpoints['ResNeXt50_32x4d']) + model.set_state_dict(paddle.load(path)) + + return model + + +def ResNeXt50_64x4d(**kwargs): + '''ResNeXt50_64x4d + ''' + pretrained = kwargs.pop('pretrained', False) + + model = _resnext.ResNeXt50_64x4d(**kwargs) + if pretrained: + assert 'ResNeXt50_64x4d' in _checkpoints, 'Not provide `ResNeXt50_64x4d` pretrained model.' + path = paddle.utils.download.get_weights_path_from_url(_checkpoints['ResNeXt50_64x4d']) + model.set_state_dict(paddle.load(path)) + + return model \ No newline at end of file