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