From c886b177822872a7f176d926f943f394e552e5ae Mon Sep 17 00:00:00 2001
From: LielinJiang <jianglielin@baidu.com>
Date: Tue, 7 Apr 2020 14:09:16 +0000
Subject: [PATCH] add more pretrained models

---
 image_classification/README.MD | 7 ++++++-
 models/mobilenetv1.py          | 6 +++++-
 models/mobilenetv2.py          | 6 +++++-
 models/vgg.py                  | 5 ++++-
 4 files changed, 20 insertions(+), 4 deletions(-)

diff --git a/image_classification/README.MD b/image_classification/README.MD
index 46ed2fa..29bd340 100644
--- a/image_classification/README.MD
+++ b/image_classification/README.MD
@@ -72,7 +72,12 @@ CUDA_VISIBLE_DEVICES=0,1,2,3 python -m paddle.distributed.launch main.py --arch
 
 | 模型 | top1 acc | top5 acc |
 | --- | --- | --- |
-| ResNet50 | 76.28 | 93.04 | 
+| [ResNet50](https://paddle-hapi.bj.bcebos.com/models/resnet50.pdparams) | 76.28 | 93.04 |
+| [vgg16](https://paddle-hapi.bj.bcebos.com/models/vgg16.pdparams) | 71.84 | 90.71 | 
+| [mobilenet_v1](https://paddle-hapi.bj.bcebos.com/models/mobilenet_v1_x1.0.pdparams) | 71.25 | 89.92 | 
+| [mobilenet_v2](https://paddle-hapi.bj.bcebos.com/models/mobilenet_v2_x1.0.pdparams) | 72.27 | 90.66 | 
+
+上述模型的复现参数请参考scripts下的脚本。
 
 
 ## 参考文献
diff --git a/models/mobilenetv1.py b/models/mobilenetv1.py
index dafcc5b..c2e7959 100644
--- a/models/mobilenetv1.py
+++ b/models/mobilenetv1.py
@@ -24,7 +24,11 @@ from .download import get_weights_path
 
 __all__ = ['MobileNetV1', 'mobilenet_v1']
 
-model_urls = {}
+model_urls = {
+    'mobilenetv1_1.0':
+    ('https://paddle-hapi.bj.bcebos.com/models/mobilenet_v1_x1.0.pdparams',
+     'bf0d25cb0bed1114d9dac9384ce2b4a6')
+}
 
 
 class ConvBNLayer(fluid.dygraph.Layer):
diff --git a/models/mobilenetv2.py b/models/mobilenetv2.py
index c665b72..0079ee7 100644
--- a/models/mobilenetv2.py
+++ b/models/mobilenetv2.py
@@ -23,7 +23,11 @@ from .download import get_weights_path
 
 __all__ = ['MobileNetV2', 'mobilenet_v2']
 
-model_urls = {}
+model_urls = {
+    'mobilenetv2_1.0':
+    ('https://paddle-hapi.bj.bcebos.com/models/mobilenet_v2_x1.0.pdparams',
+     '8ff74f291f72533f2a7956a4efff9d88')
+}
 
 
 class ConvBNLayer(fluid.dygraph.Layer):
diff --git a/models/vgg.py b/models/vgg.py
index 926c258..34f679e 100644
--- a/models/vgg.py
+++ b/models/vgg.py
@@ -32,7 +32,10 @@ __all__ = [
     'vgg19',
 ]
 
-model_urls = {}
+model_urls = {
+    'vgg16': ('https://paddle-hapi.bj.bcebos.com/models/vgg16.pdparams',
+              'c788f453a3b999063e8da043456281ee')
+}
 
 
 class Classifier(fluid.dygraph.Layer):
-- 
GitLab