add more pretrained models

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下的脚本。
 
 
 ## 参考文献

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):

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):

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):