From 04bc0cdde89d053c9d194fb79cacdc0abd5a4001 Mon Sep 17 00:00:00 2001 From: MRXLT <xlt2024@gmail.com> Date: Sun, 26 Apr 2020 14:41:00 +0800 Subject: [PATCH] add more models --- .../paddle_serving_app/models/model_list.py | 54 ++++++++++++++++++- 1 file changed, 53 insertions(+), 1 deletion(-) diff --git a/python/paddle_serving_app/models/model_list.py b/python/paddle_serving_app/models/model_list.py index 73306449..4eb109c1 100644 --- a/python/paddle_serving_app/models/model_list.py +++ b/python/paddle_serving_app/models/model_list.py @@ -28,7 +28,59 @@ class ServingModels(object): self.model_dict[ key] = "https://paddle-serving.bj.bcebos.com/paddle_hub_models/text/SentimentAnalysis/" + key + ".tar.gz" #image classification - for key in ["alexnet_imagenet"]: + for key in [ + "alexnet_imagenet", + "darknet53-imagenet", + "densenet121_imagenet", + "densenet161_imagenet", + "densenet169_imagenet", + "densenet201_imagenet", + "densenet264_imagenet" + "dpn107_imagenet", + "dpn131_imagenet", + "dpn68_imagenet", + "dpn92_imagenet", + "dpn98_imagenet", + "efficientnetb0_imagenet", + "efficientnetb1_imagenet", + "efficientnetb2_imagenet", + "efficientnetb3_imagenet", + "efficientnetb4_imagenet", + "efficientnetb5_imagenet", + "efficientnetb6_imagenet", + "googlenet_imagenet", + "inception_v4_imagenet", + "inception_v2_imagenet", + "nasnet_imagenet", + "pnasnet_imagenet", + "resnet_v2_101_imagenet", + "resnet_v2_151_imagenet", + "resnet_v2_18_imagenet", + "resnet_v2_34_imagenet", + " resnet_v2_50_imagenet", + "resnext101_32x16d_wsl", + "resnext101_32x32d_wsl", + "resnext101_32x48d_wsl", + "resnext101_32x8d_wsl", + "resnext101_32x4d_imagenet", + "resnext101_64x4d_imagenet", + "resnext101_vd_32x4d_imagenet", + "resnext101_vd_64x4d_imagenet", + "resnext152_64x4d_imagenet", + "resnext152_vd_64x4d_imagenet", + "resnext50_64x4d_imagenet", + "resnext50_vd_32x4d_imagenet", + "resnext50_vd_64x4d_imagenet", + "se_resnext101_32x4d_imagenet", + "se_resnext50_32x4d_imagenet", + "shufflenet_v2_imagenet", + "vgg11_imagenet", + "vgg13_imagenet", + "vgg16_imagenet", + "vgg19_imagenet", + "xception65_imagenet", + "xception71_imagenet", + ]: self.model_dict[ key] = "https://paddle-serving.bj.bcebos.com/paddle_hub_models/image/ImageClassification/" + key + ".tar.gz" -- GitLab