Skip to content

P
PaddleClas

PaddlePaddle / PaddleClas
大约 2 年 前同步成功

通知 118
Star 4999
Fork 1114
P
PaddleClas
提交 0102ad56 编写于 作者: L lyuwenyu
浏览文件
操作

Skip __init__ to release unnecessary dependents

上级 67a7e585
显示空白变更内容
内联 并排
Showing with 795 addition and 816 deletion
hubconf.py
浏览文件 @ 0102ad56
...@@ -12,21 +12,35 @@ ...@@ -12,21 +12,35 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
dependencies = ['paddle', 'numpy'] dependencies = ['paddle']
import paddle import paddle
from ppcls.modeling import architectures import os
import sys
def _load_pretrained_parameters(model, name): class _SysPathG(object):
def __enter__(self, ):
sys.path.insert(0,
os.path.join(
os.path.dirname(os.path.abspath(__file__)),
'ppcls', 'modeling'))
def __exit__(self, type, value, traceback):
sys.path.pop(0)
with _SysPathG():
import architectures
def _load_pretrained_parameters(model, name):
url = 'https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/{}_pretrained.pdparams'.format( url = 'https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/{}_pretrained.pdparams'.format(
name) name)
path = paddle.utils.download.get_weights_path_from_url(url) path = paddle.utils.download.get_weights_path_from_url(url)
model.set_state_dict(paddle.load(path)) model.set_state_dict(paddle.load(path))
return model return model
def alexnet(pretrained=False, **kwargs):
def alexnet(pretrained=False, **kwargs):
""" """
AlexNet AlexNet
Args: Args:
...@@ -42,8 +56,7 @@ def alexnet(pretrained=False, **kwargs): ...@@ -42,8 +56,7 @@ def alexnet(pretrained=False, **kwargs):
return model return model
def vgg11(pretrained=False, **kwargs):
def vgg11(pretrained=False, **kwargs):
""" """
VGG11 VGG11
Args: Args:
...@@ -60,8 +73,7 @@ def vgg11(pretrained=False, **kwargs): ...@@ -60,8 +73,7 @@ def vgg11(pretrained=False, **kwargs):
return model return model
def vgg13(pretrained=False, **kwargs):
def vgg13(pretrained=False, **kwargs):
""" """
VGG13 VGG13
Args: Args:
...@@ -78,8 +90,7 @@ def vgg13(pretrained=False, **kwargs): ...@@ -78,8 +90,7 @@ def vgg13(pretrained=False, **kwargs):
return model return model
def vgg16(pretrained=False, **kwargs):
def vgg16(pretrained=False, **kwargs):
""" """
VGG16 VGG16
Args: Args:
...@@ -96,8 +107,7 @@ def vgg16(pretrained=False, **kwargs): ...@@ -96,8 +107,7 @@ def vgg16(pretrained=False, **kwargs):
return model return model
def vgg19(pretrained=False, **kwargs):
def vgg19(pretrained=False, **kwargs):
""" """
VGG19 VGG19
Args: Args:
...@@ -114,8 +124,7 @@ def vgg19(pretrained=False, **kwargs): ...@@ -114,8 +124,7 @@ def vgg19(pretrained=False, **kwargs):
return model return model
def resnet18(pretrained=False, **kwargs):
def resnet18(pretrained=False, **kwargs):
""" """
ResNet18 ResNet18
Args: Args:
...@@ -133,8 +142,7 @@ def resnet18(pretrained=False, **kwargs): ...@@ -133,8 +142,7 @@ def resnet18(pretrained=False, **kwargs):
return model return model
def resnet34(pretrained=False, **kwargs):
def resnet34(pretrained=False, **kwargs):
""" """
ResNet34 ResNet34
Args: Args:
...@@ -152,8 +160,7 @@ def resnet34(pretrained=False, **kwargs): ...@@ -152,8 +160,7 @@ def resnet34(pretrained=False, **kwargs):
return model return model
def resnet50(pretrained=False, **kwargs):
def resnet50(pretrained=False, **kwargs):
""" """
ResNet50 ResNet50
Args: Args:
...@@ -171,8 +178,7 @@ def resnet50(pretrained=False, **kwargs): ...@@ -171,8 +178,7 @@ def resnet50(pretrained=False, **kwargs):
return model return model
def resnet101(pretrained=False, **kwargs):
def resnet101(pretrained=False, **kwargs):
""" """
ResNet101 ResNet101
Args: Args:
...@@ -190,8 +196,7 @@ def resnet101(pretrained=False, **kwargs): ...@@ -190,8 +196,7 @@ def resnet101(pretrained=False, **kwargs):
return model return model
def resnet152(pretrained=False, **kwargs):
def resnet152(pretrained=False, **kwargs):
""" """
ResNet152 ResNet152
Args: Args:
...@@ -209,8 +214,7 @@ def resnet152(pretrained=False, **kwargs): ...@@ -209,8 +214,7 @@ def resnet152(pretrained=False, **kwargs):
return model return model
def squeezenet1_0(pretrained=False, **kwargs):
def squeezenet1_0(pretrained=False, **kwargs):
""" """
SqueezeNet1_0 SqueezeNet1_0
Args: Args:
...@@ -226,8 +230,7 @@ def squeezenet1_0(pretrained=False, **kwargs): ...@@ -226,8 +230,7 @@ def squeezenet1_0(pretrained=False, **kwargs):
return model return model
def squeezenet1_1(pretrained=False, **kwargs):
def squeezenet1_1(pretrained=False, **kwargs):
""" """
SqueezeNet1_1 SqueezeNet1_1
Args: Args:
...@@ -243,8 +246,7 @@ def squeezenet1_1(pretrained=False, **kwargs): ...@@ -243,8 +246,7 @@ def squeezenet1_1(pretrained=False, **kwargs):
return model return model
def densenet121(pretrained=False, **kwargs):
def densenet121(pretrained=False, **kwargs):
""" """
DenseNet121 DenseNet121
Args: Args:
...@@ -262,8 +264,7 @@ def densenet121(pretrained=False, **kwargs): ...@@ -262,8 +264,7 @@ def densenet121(pretrained=False, **kwargs):
return model return model
def densenet161(pretrained=False, **kwargs):
def densenet161(pretrained=False, **kwargs):
""" """
DenseNet161 DenseNet161
Args: Args:
...@@ -281,8 +282,7 @@ def densenet161(pretrained=False, **kwargs): ...@@ -281,8 +282,7 @@ def densenet161(pretrained=False, **kwargs):
return model return model
def densenet169(pretrained=False, **kwargs):
def densenet169(pretrained=False, **kwargs):
""" """
DenseNet169 DenseNet169
Args: Args:
...@@ -300,8 +300,7 @@ def densenet169(pretrained=False, **kwargs): ...@@ -300,8 +300,7 @@ def densenet169(pretrained=False, **kwargs):
return model return model
def densenet201(pretrained=False, **kwargs):
def densenet201(pretrained=False, **kwargs):
""" """
DenseNet201 DenseNet201
Args: Args:
...@@ -319,8 +318,7 @@ def densenet201(pretrained=False, **kwargs): ...@@ -319,8 +318,7 @@ def densenet201(pretrained=False, **kwargs):
return model return model
def densenet264(pretrained=False, **kwargs):
def densenet264(pretrained=False, **kwargs):
""" """
DenseNet264 DenseNet264
Args: Args:
...@@ -338,8 +336,7 @@ def densenet264(pretrained=False, **kwargs): ...@@ -338,8 +336,7 @@ def densenet264(pretrained=False, **kwargs):
return model return model
def inceptionv3(pretrained=False, **kwargs):
def inceptionv3(pretrained=False, **kwargs):
""" """
InceptionV3 InceptionV3
Args: Args:
...@@ -355,8 +352,7 @@ def inceptionv3(pretrained=False, **kwargs): ...@@ -355,8 +352,7 @@ def inceptionv3(pretrained=False, **kwargs):
return model return model
def inceptionv4(pretrained=False, **kwargs):
def inceptionv4(pretrained=False, **kwargs):
""" """
InceptionV4 InceptionV4
Args: Args:
...@@ -372,8 +368,7 @@ def inceptionv4(pretrained=False, **kwargs): ...@@ -372,8 +368,7 @@ def inceptionv4(pretrained=False, **kwargs):
return model return model
def googlenet(pretrained=False, **kwargs):
def googlenet(pretrained=False, **kwargs):
""" """
GoogLeNet GoogLeNet
Args: Args:
...@@ -389,8 +384,7 @@ def googlenet(pretrained=False, **kwargs): ...@@ -389,8 +384,7 @@ def googlenet(pretrained=False, **kwargs):
return model return model
def shufflenetv2_x0_25(pretrained=False, **kwargs):
def shufflenetv2_x0_25(pretrained=False, **kwargs):
""" """
ShuffleNetV2_x0_25 ShuffleNetV2_x0_25
Args: Args:
...@@ -406,8 +400,7 @@ def shufflenetv2_x0_25(pretrained=False, **kwargs): ...@@ -406,8 +400,7 @@ def shufflenetv2_x0_25(pretrained=False, **kwargs):
return model return model
def mobilenetv1(pretrained=False, **kwargs):
def mobilenetv1(pretrained=False, **kwargs):
""" """
MobileNetV1 MobileNetV1
Args: Args:
...@@ -423,8 +416,7 @@ def mobilenetv1(pretrained=False, **kwargs): ...@@ -423,8 +416,7 @@ def mobilenetv1(pretrained=False, **kwargs):
return model return model
def mobilenetv1_x0_25(pretrained=False, **kwargs):
def mobilenetv1_x0_25(pretrained=False, **kwargs):
""" """
MobileNetV1_x0_25 MobileNetV1_x0_25
Args: Args:
...@@ -440,8 +432,7 @@ def mobilenetv1_x0_25(pretrained=False, **kwargs): ...@@ -440,8 +432,7 @@ def mobilenetv1_x0_25(pretrained=False, **kwargs):
return model return model
def mobilenetv1_x0_5(pretrained=False, **kwargs):
def mobilenetv1_x0_5(pretrained=False, **kwargs):
""" """
MobileNetV1_x0_5 MobileNetV1_x0_5
Args: Args:
...@@ -457,8 +448,7 @@ def mobilenetv1_x0_5(pretrained=False, **kwargs): ...@@ -457,8 +448,7 @@ def mobilenetv1_x0_5(pretrained=False, **kwargs):
return model return model
def mobilenetv1_x0_75(pretrained=False, **kwargs):
def mobilenetv1_x0_75(pretrained=False, **kwargs):
""" """
MobileNetV1_x0_75 MobileNetV1_x0_75
Args: Args:
...@@ -474,8 +464,7 @@ def mobilenetv1_x0_75(pretrained=False, **kwargs): ...@@ -474,8 +464,7 @@ def mobilenetv1_x0_75(pretrained=False, **kwargs):
return model return model
def mobilenetv2_x0_25(pretrained=False, **kwargs):
def mobilenetv2_x0_25(pretrained=False, **kwargs):
""" """
MobileNetV2_x0_25 MobileNetV2_x0_25
Args: Args:
...@@ -491,8 +480,7 @@ def mobilenetv2_x0_25(pretrained=False, **kwargs): ...@@ -491,8 +480,7 @@ def mobilenetv2_x0_25(pretrained=False, **kwargs):
return model return model
def mobilenetv2_x0_5(pretrained=False, **kwargs):
def mobilenetv2_x0_5(pretrained=False, **kwargs):
""" """
MobileNetV2_x0_5 MobileNetV2_x0_5
Args: Args:
...@@ -508,8 +496,7 @@ def mobilenetv2_x0_5(pretrained=False, **kwargs): ...@@ -508,8 +496,7 @@ def mobilenetv2_x0_5(pretrained=False, **kwargs):
return model return model
def mobilenetv2_x0_75(pretrained=False, **kwargs):
def mobilenetv2_x0_75(pretrained=False, **kwargs):
""" """
MobileNetV2_x0_75 MobileNetV2_x0_75
Args: Args:
...@@ -525,8 +512,7 @@ def mobilenetv2_x0_75(pretrained=False, **kwargs): ...@@ -525,8 +512,7 @@ def mobilenetv2_x0_75(pretrained=False, **kwargs):
return model return model
def mobilenetv2_x1_5(pretrained=False, **kwargs):
def mobilenetv2_x1_5(pretrained=False, **kwargs):
""" """
MobileNetV2_x1_5 MobileNetV2_x1_5
Args: Args:
...@@ -542,8 +528,7 @@ def mobilenetv2_x1_5(pretrained=False, **kwargs): ...@@ -542,8 +528,7 @@ def mobilenetv2_x1_5(pretrained=False, **kwargs):
return model return model
def mobilenetv2_x2_0(pretrained=False, **kwargs):
def mobilenetv2_x2_0(pretrained=False, **kwargs):
""" """
MobileNetV2_x2_0 MobileNetV2_x2_0
Args: Args:
...@@ -559,8 +544,7 @@ def mobilenetv2_x2_0(pretrained=False, **kwargs): ...@@ -559,8 +544,7 @@ def mobilenetv2_x2_0(pretrained=False, **kwargs):
return model return model
def mobilenetv3_large_x0_35(pretrained=False, **kwargs):
def mobilenetv3_large_x0_35(pretrained=False, **kwargs):
""" """
MobileNetV3_large_x0_35 MobileNetV3_large_x0_35
Args: Args:
...@@ -572,12 +556,12 @@ def mobilenetv3_large_x0_35(pretrained=False, **kwargs): ...@@ -572,12 +556,12 @@ def mobilenetv3_large_x0_35(pretrained=False, **kwargs):
""" """
model = architectures.MobileNetV3_large_x0_35(**kwargs) model = architectures.MobileNetV3_large_x0_35(**kwargs)
if pretrained: if pretrained:
model = _load_pretrained_parameters(model, 'MobileNetV3_large_x0_35') model = _load_pretrained_parameters(model,
'MobileNetV3_large_x0_35')
return model return model
def mobilenetv3_large_x0_5(pretrained=False, **kwargs):
def mobilenetv3_large_x0_5(pretrained=False, **kwargs):
""" """
MobileNetV3_large_x0_5 MobileNetV3_large_x0_5
Args: Args:
...@@ -589,12 +573,12 @@ def mobilenetv3_large_x0_5(pretrained=False, **kwargs): ...@@ -589,12 +573,12 @@ def mobilenetv3_large_x0_5(pretrained=False, **kwargs):
""" """
model = architectures.MobileNetV3_large_x0_5(**kwargs) model = architectures.MobileNetV3_large_x0_5(**kwargs)
if pretrained: if pretrained:
model = _load_pretrained_parameters(model, 'MobileNetV3_large_x0_5') model = _load_pretrained_parameters(model,
'MobileNetV3_large_x0_5')
return model return model
def mobilenetv3_large_x0_75(pretrained=False, **kwargs):
def mobilenetv3_large_x0_75(pretrained=False, **kwargs):
""" """
MobileNetV3_large_x0_75 MobileNetV3_large_x0_75
Args: Args:
...@@ -606,12 +590,12 @@ def mobilenetv3_large_x0_75(pretrained=False, **kwargs): ...@@ -606,12 +590,12 @@ def mobilenetv3_large_x0_75(pretrained=False, **kwargs):
""" """
model = architectures.MobileNetV3_large_x0_75(**kwargs) model = architectures.MobileNetV3_large_x0_75(**kwargs)
if pretrained: if pretrained:
model = _load_pretrained_parameters(model, 'MobileNetV3_large_x0_75') model = _load_pretrained_parameters(model,
'MobileNetV3_large_x0_75')
return model return model
def mobilenetv3_large_x1_0(pretrained=False, **kwargs):
def mobilenetv3_large_x1_0(pretrained=False, **kwargs):
""" """
MobileNetV3_large_x1_0 MobileNetV3_large_x1_0
Args: Args:
...@@ -623,12 +607,12 @@ def mobilenetv3_large_x1_0(pretrained=False, **kwargs): ...@@ -623,12 +607,12 @@ def mobilenetv3_large_x1_0(pretrained=False, **kwargs):
""" """
model = architectures.MobileNetV3_large_x1_0(**kwargs) model = architectures.MobileNetV3_large_x1_0(**kwargs)
if pretrained: if pretrained:
model = _load_pretrained_parameters(model, 'MobileNetV3_large_x1_0') model = _load_pretrained_parameters(model,
'MobileNetV3_large_x1_0')
return model return model
def mobilenetv3_large_x1_25(pretrained=False, **kwargs):
def mobilenetv3_large_x1_25(pretrained=False, **kwargs):
""" """
MobileNetV3_large_x1_25 MobileNetV3_large_x1_25
Args: Args:
...@@ -640,12 +624,12 @@ def mobilenetv3_large_x1_25(pretrained=False, **kwargs): ...@@ -640,12 +624,12 @@ def mobilenetv3_large_x1_25(pretrained=False, **kwargs):
""" """
model = architectures.MobileNetV3_large_x1_25(**kwargs) model = architectures.MobileNetV3_large_x1_25(**kwargs)
if pretrained: if pretrained:
model = _load_pretrained_parameters(model, 'MobileNetV3_large_x1_25') model = _load_pretrained_parameters(model,
'MobileNetV3_large_x1_25')
return model return model
def mobilenetv3_small_x0_35(pretrained=False, **kwargs):
def mobilenetv3_small_x0_35(pretrained=False, **kwargs):
""" """
MobileNetV3_small_x0_35 MobileNetV3_small_x0_35
Args: Args:
...@@ -657,12 +641,12 @@ def mobilenetv3_small_x0_35(pretrained=False, **kwargs): ...@@ -657,12 +641,12 @@ def mobilenetv3_small_x0_35(pretrained=False, **kwargs):
""" """
model = architectures.MobileNetV3_small_x0_35(**kwargs) model = architectures.MobileNetV3_small_x0_35(**kwargs)
if pretrained: if pretrained:
model = _load_pretrained_parameters(model, 'MobileNetV3_small_x0_35') model = _load_pretrained_parameters(model,
'MobileNetV3_small_x0_35')
return model return model
def mobilenetv3_small_x0_5(pretrained=False, **kwargs):
def mobilenetv3_small_x0_5(pretrained=False, **kwargs):
""" """
MobileNetV3_small_x0_5 MobileNetV3_small_x0_5
Args: Args:
...@@ -674,12 +658,12 @@ def mobilenetv3_small_x0_5(pretrained=False, **kwargs): ...@@ -674,12 +658,12 @@ def mobilenetv3_small_x0_5(pretrained=False, **kwargs):
""" """
model = architectures.MobileNetV3_small_x0_5(**kwargs) model = architectures.MobileNetV3_small_x0_5(**kwargs)
if pretrained: if pretrained:
model = _load_pretrained_parameters(model, 'MobileNetV3_small_x0_5') model = _load_pretrained_parameters(model,
'MobileNetV3_small_x0_5')
return model return model
def mobilenetv3_small_x0_75(pretrained=False, **kwargs):
def mobilenetv3_small_x0_75(pretrained=False, **kwargs):
""" """
MobileNetV3_small_x0_75 MobileNetV3_small_x0_75
Args: Args:
...@@ -691,12 +675,12 @@ def mobilenetv3_small_x0_75(pretrained=False, **kwargs): ...@@ -691,12 +675,12 @@ def mobilenetv3_small_x0_75(pretrained=False, **kwargs):
""" """
model = architectures.MobileNetV3_small_x0_75(**kwargs) model = architectures.MobileNetV3_small_x0_75(**kwargs)
if pretrained: if pretrained:
model = _load_pretrained_parameters(model, 'MobileNetV3_small_x0_75') model = _load_pretrained_parameters(model,
'MobileNetV3_small_x0_75')
return model return model
def mobilenetv3_small_x1_0(pretrained=False, **kwargs):
def mobilenetv3_small_x1_0(pretrained=False, **kwargs):
""" """
MobileNetV3_small_x1_0 MobileNetV3_small_x1_0
Args: Args:
...@@ -708,12 +692,12 @@ def mobilenetv3_small_x1_0(pretrained=False, **kwargs): ...@@ -708,12 +692,12 @@ def mobilenetv3_small_x1_0(pretrained=False, **kwargs):
""" """
model = architectures.MobileNetV3_small_x1_0(**kwargs) model = architectures.MobileNetV3_small_x1_0(**kwargs)
if pretrained: if pretrained:
model = _load_pretrained_parameters(model, 'MobileNetV3_small_x1_0') model = _load_pretrained_parameters(model,
'MobileNetV3_small_x1_0')
return model return model
def mobilenetv3_small_x1_25(pretrained=False, **kwargs):
def mobilenetv3_small_x1_25(pretrained=False, **kwargs):
""" """
MobileNetV3_small_x1_25 MobileNetV3_small_x1_25
Args: Args:
...@@ -725,12 +709,12 @@ def mobilenetv3_small_x1_25(pretrained=False, **kwargs): ...@@ -725,12 +709,12 @@ def mobilenetv3_small_x1_25(pretrained=False, **kwargs):
""" """
model = architectures.MobileNetV3_small_x1_25(**kwargs) model = architectures.MobileNetV3_small_x1_25(**kwargs)
if pretrained: if pretrained:
model = _load_pretrained_parameters(model, 'MobileNetV3_small_x1_25') model = _load_pretrained_parameters(model,
'MobileNetV3_small_x1_25')
return model return model
def resnext101_32x4d(pretrained=False, **kwargs):
def resnext101_32x4d(pretrained=False, **kwargs):
""" """
ResNeXt101_32x4d ResNeXt101_32x4d
Args: Args:
...@@ -746,8 +730,7 @@ def resnext101_32x4d(pretrained=False, **kwargs): ...@@ -746,8 +730,7 @@ def resnext101_32x4d(pretrained=False, **kwargs):
return model return model
def resnext101_64x4d(pretrained=False, **kwargs):
def resnext101_64x4d(pretrained=False, **kwargs):
""" """
ResNeXt101_64x4d ResNeXt101_64x4d
Args: Args:
...@@ -763,8 +746,7 @@ def resnext101_64x4d(pretrained=False, **kwargs): ...@@ -763,8 +746,7 @@ def resnext101_64x4d(pretrained=False, **kwargs):
return model return model
def resnext152_32x4d(pretrained=False, **kwargs):
def resnext152_32x4d(pretrained=False, **kwargs):
""" """
ResNeXt152_32x4d ResNeXt152_32x4d
Args: Args:
...@@ -780,8 +762,7 @@ def resnext152_32x4d(pretrained=False, **kwargs): ...@@ -780,8 +762,7 @@ def resnext152_32x4d(pretrained=False, **kwargs):
return model return model
def resnext152_64x4d(pretrained=False, **kwargs):
def resnext152_64x4d(pretrained=False, **kwargs):
""" """
ResNeXt152_64x4d ResNeXt152_64x4d
Args: Args:
...@@ -797,8 +778,7 @@ def resnext152_64x4d(pretrained=False, **kwargs): ...@@ -797,8 +778,7 @@ def resnext152_64x4d(pretrained=False, **kwargs):
return model return model
def resnext50_32x4d(pretrained=False, **kwargs):
def resnext50_32x4d(pretrained=False, **kwargs):
""" """
ResNeXt50_32x4d ResNeXt50_32x4d
Args: Args:
...@@ -814,8 +794,7 @@ def resnext50_32x4d(pretrained=False, **kwargs): ...@@ -814,8 +794,7 @@ def resnext50_32x4d(pretrained=False, **kwargs):
return model return model
def resnext50_64x4d(pretrained=False, **kwargs):
def resnext50_64x4d(pretrained=False, **kwargs):
""" """
ResNeXt50_64x4d ResNeXt50_64x4d
Args: Args:
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或