From 124a8a68d47608b0e9750f64a4af33b685c2d76f Mon Sep 17 00:00:00 2001 From: Guanghua Yu <742925032@qq.com> Date: Thu, 14 Jan 2021 19:17:13 +0800 Subject: [PATCH] refine dygraph config (#2051) * refine dygraph config, test=dygraph * fix fcos config --- .../cascade_mask_rcnn_r50_fpn_1x_coco.yml | 7 --- dygraph/configs/cascade_rcnn/README.md | 23 ++++++++ .../_base_/cascade_fpn_reader.yml} | 0 .../_base_/cascade_mask_fpn_reader.yml} | 0 .../_base_}/cascade_mask_rcnn_r50_fpn.yml | 1 - .../_base_}/cascade_rcnn_r50_fpn.yml | 1 - .../_base_/optimizer_1x.yml} | 0 .../cascade_mask_rcnn_r50_fpn_1x_coco.yml | 8 +++ .../cascade_rcnn_r50_fpn_1x_coco.yml | 8 +++ .../configs/cascade_rcnn_r50_fpn_1x_coco.yml | 7 --- .../{_base_ => }/datasets/coco_detection.yml | 0 .../{_base_ => }/datasets/coco_instance.yml | 0 dygraph/configs/{_base_ => }/datasets/voc.yml | 0 dygraph/configs/faster_rcnn/README.md | 20 +++++++ .../faster_rcnn/_base_/faster_fpn_reader.yml | 40 +++++++++++++ .../_base_}/faster_rcnn_r50.yml | 1 - .../_base_}/faster_rcnn_r50_fpn.yml | 1 - .../_base_}/faster_reader.yml | 0 .../faster_rcnn/_base_/optimizer_1x.yml | 19 ++++++ .../faster_rcnn/faster_rcnn_r50_1x_coco.yml | 8 +++ .../faster_rcnn_r50_fpn_1x_coco.yml | 8 +++ dygraph/configs/faster_rcnn_r50_1x_coco.yml | 7 --- .../configs/faster_rcnn_r50_fpn_1x_coco.yml | 7 --- dygraph/configs/fcos/fcos_r50_fpn_1x_coco.yml | 4 +- dygraph/configs/mask_rcnn/README.md | 20 +++++++ .../mask_rcnn/_base_/mask_fpn_reader.yml | 40 +++++++++++++ .../_base_}/mask_rcnn_r50.yml | 1 - .../_base_}/mask_rcnn_r50_fpn.yml | 1 - .../_base_}/mask_reader.yml | 0 .../configs/mask_rcnn/_base_/optimizer_1x.yml | 19 ++++++ .../mask_rcnn/mask_rcnn_r50_1x_coco.yml | 8 +++ .../mask_rcnn/mask_rcnn_r50_fpn_1x_coco.yml | 8 +++ dygraph/configs/mask_rcnn_r50_1x_coco.yml | 7 --- dygraph/configs/mask_rcnn_r50_fpn_1x_coco.yml | 7 --- dygraph/configs/{_base_ => }/runtime.yml | 0 dygraph/configs/ssd/README.md | 21 +++++++ .../_base_/optimizer_1000e.yml} | 0 .../_base_/optimizer_120e.yml} | 0 .../_base_/optimizer_240e.yml} | 0 .../_base_}/ssd_mobilenet_reader.yml | 0 .../_base_}/ssd_mobilenet_v1_300.yml | 2 +- .../readers => ssd/_base_}/ssd_reader.yml | 0 .../models => ssd/_base_}/ssd_vgg16_300.yml | 2 +- .../_base_}/ssdlite300_reader.yml | 0 .../_base_}/ssdlite320_reader.yml | 0 .../_base_}/ssdlite_mobilenet_v1_300.yml | 2 +- .../ssdlite_mobilenet_v3_large_320.yml | 2 +- .../ssdlite_mobilenet_v3_small_320.yml | 2 +- .../ssd/ssd_mobilenet_v1_300_120e_voc.yml | 8 +++ .../configs/ssd/ssd_vgg16_300_240e_voc.yml | 8 +++ .../ssd/ssdlite_mobilenet_v1_300_coco.yml | 8 +++ .../ssdlite_mobilenet_v3_large_320_coco.yml | 8 +++ .../ssdlite_mobilenet_v3_small_320_coco.yml | 8 +++ .../configs/ssd_mobilenet_v1_300_120e_voc.yml | 7 --- dygraph/configs/ssd_vgg16_300_240e_voc.yml | 7 --- .../configs/ssdlite_mobilenet_v1_300_coco.yml | 7 --- .../ssdlite_mobilenet_v3_large_320_coco.yml | 7 --- .../ssdlite_mobilenet_v3_small_320_coco.yml | 7 --- dygraph/configs/yolov3/README.md | 46 +++++++++++++++ .../_base_/optimizer_270e.yml} | 0 .../_base_}/yolov3_darknet53.yml | 1 - .../_base_}/yolov3_mobilenet_v1.yml | 1 - .../_base_}/yolov3_mobilenet_v3_large.yml | 1 - .../_base_}/yolov3_mobilenet_v3_small.yml | 1 - .../_base_}/yolov3_reader.yml | 0 .../yolov3/yolov3_darknet53_270e_coco.yml | 8 +++ .../yolov3/yolov3_mobilenet_v1_270e_coco.yml | 8 +++ .../yolov3_mobilenet_v1_270e_voc.yml | 11 ++-- .../yolov3_mobilenet_v3_large_270e_coco.yml | 8 +++ .../yolov3_mobilenet_v3_large_270e_voc.yml | 11 ++-- .../configs/yolov3_darknet53_270e_coco.yml | 7 --- .../configs/yolov3_mobilenet_v1_270e_coco.yml | 7 --- .../yolov3_mobilenet_v3_large_270e_coco.yml | 7 --- dygraph/docs/MODEL_ZOO_cn.md | 58 +++++-------------- 74 files changed, 392 insertions(+), 170 deletions(-) delete mode 100644 dygraph/configs/cascade_mask_rcnn_r50_fpn_1x_coco.yml create mode 100644 dygraph/configs/cascade_rcnn/README.md rename dygraph/configs/{_base_/readers/faster_fpn_reader.yml => cascade_rcnn/_base_/cascade_fpn_reader.yml} (100%) rename dygraph/configs/{_base_/readers/mask_fpn_reader.yml => cascade_rcnn/_base_/cascade_mask_fpn_reader.yml} (100%) rename dygraph/configs/{_base_/models => cascade_rcnn/_base_}/cascade_mask_rcnn_r50_fpn.yml (97%) rename dygraph/configs/{_base_/models => cascade_rcnn/_base_}/cascade_rcnn_r50_fpn.yml (97%) rename dygraph/configs/{_base_/optimizers/rcnn_1x.yml => cascade_rcnn/_base_/optimizer_1x.yml} (100%) create mode 100644 dygraph/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco.yml create mode 100644 dygraph/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco.yml delete mode 100644 dygraph/configs/cascade_rcnn_r50_fpn_1x_coco.yml rename dygraph/configs/{_base_ => }/datasets/coco_detection.yml (100%) rename dygraph/configs/{_base_ => }/datasets/coco_instance.yml (100%) rename dygraph/configs/{_base_ => }/datasets/voc.yml (100%) create mode 100644 dygraph/configs/faster_rcnn/README.md create mode 100644 dygraph/configs/faster_rcnn/_base_/faster_fpn_reader.yml rename dygraph/configs/{_base_/models => faster_rcnn/_base_}/faster_rcnn_r50.yml (96%) rename dygraph/configs/{_base_/models => faster_rcnn/_base_}/faster_rcnn_r50_fpn.yml (96%) rename dygraph/configs/{_base_/readers => faster_rcnn/_base_}/faster_reader.yml (100%) create mode 100644 dygraph/configs/faster_rcnn/_base_/optimizer_1x.yml create mode 100644 dygraph/configs/faster_rcnn/faster_rcnn_r50_1x_coco.yml create mode 100644 dygraph/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.yml delete mode 100644 dygraph/configs/faster_rcnn_r50_1x_coco.yml delete mode 100644 dygraph/configs/faster_rcnn_r50_fpn_1x_coco.yml create mode 100644 dygraph/configs/mask_rcnn/README.md create mode 100644 dygraph/configs/mask_rcnn/_base_/mask_fpn_reader.yml rename dygraph/configs/{_base_/models => mask_rcnn/_base_}/mask_rcnn_r50.yml (97%) rename dygraph/configs/{_base_/models => mask_rcnn/_base_}/mask_rcnn_r50_fpn.yml (97%) rename dygraph/configs/{_base_/readers => mask_rcnn/_base_}/mask_reader.yml (100%) create mode 100644 dygraph/configs/mask_rcnn/_base_/optimizer_1x.yml create mode 100644 dygraph/configs/mask_rcnn/mask_rcnn_r50_1x_coco.yml create mode 100644 dygraph/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco.yml delete mode 100644 dygraph/configs/mask_rcnn_r50_1x_coco.yml delete mode 100644 dygraph/configs/mask_rcnn_r50_fpn_1x_coco.yml rename dygraph/configs/{_base_ => }/runtime.yml (100%) create mode 100644 dygraph/configs/ssd/README.md rename dygraph/configs/{_base_/optimizers/ssdlite_1000e.yml => ssd/_base_/optimizer_1000e.yml} (100%) rename dygraph/configs/{_base_/optimizers/ssd_mobilenet_120e.yml => ssd/_base_/optimizer_120e.yml} (100%) rename dygraph/configs/{_base_/optimizers/ssd_240e.yml => ssd/_base_/optimizer_240e.yml} (100%) rename dygraph/configs/{_base_/readers => ssd/_base_}/ssd_mobilenet_reader.yml (100%) rename dygraph/configs/{_base_/models => ssd/_base_}/ssd_mobilenet_v1_300.yml (96%) rename dygraph/configs/{_base_/readers => ssd/_base_}/ssd_reader.yml (100%) rename dygraph/configs/{_base_/models => ssd/_base_}/ssd_vgg16_300.yml (96%) rename dygraph/configs/{_base_/readers => ssd/_base_}/ssdlite300_reader.yml (100%) rename dygraph/configs/{_base_/readers => ssd/_base_}/ssdlite320_reader.yml (100%) rename dygraph/configs/{_base_/models => ssd/_base_}/ssdlite_mobilenet_v1_300.yml (95%) rename dygraph/configs/{_base_/models => ssd/_base_}/ssdlite_mobilenet_v3_large_320.yml (95%) rename dygraph/configs/{_base_/models => ssd/_base_}/ssdlite_mobilenet_v3_small_320.yml (95%) create mode 100644 dygraph/configs/ssd/ssd_mobilenet_v1_300_120e_voc.yml create mode 100644 dygraph/configs/ssd/ssd_vgg16_300_240e_voc.yml create mode 100644 dygraph/configs/ssd/ssdlite_mobilenet_v1_300_coco.yml create mode 100644 dygraph/configs/ssd/ssdlite_mobilenet_v3_large_320_coco.yml create mode 100644 dygraph/configs/ssd/ssdlite_mobilenet_v3_small_320_coco.yml delete mode 100644 dygraph/configs/ssd_mobilenet_v1_300_120e_voc.yml delete mode 100644 dygraph/configs/ssd_vgg16_300_240e_voc.yml delete mode 100644 dygraph/configs/ssdlite_mobilenet_v1_300_coco.yml delete mode 100644 dygraph/configs/ssdlite_mobilenet_v3_large_320_coco.yml delete mode 100644 dygraph/configs/ssdlite_mobilenet_v3_small_320_coco.yml create mode 100644 dygraph/configs/yolov3/README.md rename dygraph/configs/{_base_/optimizers/yolov3_270e.yml => yolov3/_base_/optimizer_270e.yml} (100%) rename dygraph/configs/{_base_/models => yolov3/_base_}/yolov3_darknet53.yml (95%) rename dygraph/configs/{_base_/models => yolov3/_base_}/yolov3_mobilenet_v1.yml (95%) rename dygraph/configs/{_base_/models => yolov3/_base_}/yolov3_mobilenet_v3_large.yml (95%) rename dygraph/configs/{_base_/models => yolov3/_base_}/yolov3_mobilenet_v3_small.yml (95%) rename dygraph/configs/{_base_/readers => yolov3/_base_}/yolov3_reader.yml (100%) create mode 100644 dygraph/configs/yolov3/yolov3_darknet53_270e_coco.yml create mode 100644 dygraph/configs/yolov3/yolov3_mobilenet_v1_270e_coco.yml rename dygraph/configs/{ => yolov3}/yolov3_mobilenet_v1_270e_voc.yml (85%) create mode 100644 dygraph/configs/yolov3/yolov3_mobilenet_v3_large_270e_coco.yml rename dygraph/configs/{ => yolov3}/yolov3_mobilenet_v3_large_270e_voc.yml (85%) delete mode 100644 dygraph/configs/yolov3_darknet53_270e_coco.yml delete mode 100644 dygraph/configs/yolov3_mobilenet_v1_270e_coco.yml delete mode 100644 dygraph/configs/yolov3_mobilenet_v3_large_270e_coco.yml diff --git a/dygraph/configs/cascade_mask_rcnn_r50_fpn_1x_coco.yml b/dygraph/configs/cascade_mask_rcnn_r50_fpn_1x_coco.yml deleted file mode 100644 index b38bd32f5..000000000 --- a/dygraph/configs/cascade_mask_rcnn_r50_fpn_1x_coco.yml +++ /dev/null @@ -1,7 +0,0 @@ -_BASE_: [ - './_base_/models/cascade_mask_rcnn_r50_fpn.yml', - './_base_/optimizers/rcnn_1x.yml', - './_base_/datasets/coco_instance.yml', - './_base_/readers/mask_fpn_reader.yml', - './_base_/runtime.yml', -] diff --git a/dygraph/configs/cascade_rcnn/README.md b/dygraph/configs/cascade_rcnn/README.md new file mode 100644 index 000000000..aac21739c --- /dev/null +++ b/dygraph/configs/cascade_rcnn/README.md @@ -0,0 +1,23 @@ +# Cascade R-CNN: High Quality Object Detection and Instance Segmentation + +## Model Zoo + +| 骨架网络 | 网络类型 | 每张GPU图片个数 | 学习率策略 |推理时间(fps) | Box AP | Mask AP | 下载 | 配置文件 | +| :------------------- | :------------- | :-----: | :-----: | :------------: | :-----: | :-----: | :-----------------------------------------------------: | :-----: | +| ResNet50-FPN | Cascade Faster | 1 | 1x | ---- | 41.1 | - | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/cascade_rcnn_r50_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/cascade_faster_rcnn_r50_fpn_1x_coco.yml) | +| ResNet50-FPN | Cascade Mask | 1 | 1x | ---- | 41.6 | 35.3 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/cascade_mask_rcnn_r50_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/cascade_mask_rcnn_r50_fpn_1x_coco.yml) | + +## Citations +``` +@article{Cai_2019, + title={Cascade R-CNN: High Quality Object Detection and Instance Segmentation}, + ISSN={1939-3539}, + url={http://dx.doi.org/10.1109/tpami.2019.2956516}, + DOI={10.1109/tpami.2019.2956516}, + journal={IEEE Transactions on Pattern Analysis and Machine Intelligence}, + publisher={Institute of Electrical and Electronics Engineers (IEEE)}, + author={Cai, Zhaowei and Vasconcelos, Nuno}, + year={2019}, + pages={1–1} +} +``` diff --git a/dygraph/configs/_base_/readers/faster_fpn_reader.yml b/dygraph/configs/cascade_rcnn/_base_/cascade_fpn_reader.yml similarity index 100% rename from dygraph/configs/_base_/readers/faster_fpn_reader.yml rename to dygraph/configs/cascade_rcnn/_base_/cascade_fpn_reader.yml diff --git a/dygraph/configs/_base_/readers/mask_fpn_reader.yml b/dygraph/configs/cascade_rcnn/_base_/cascade_mask_fpn_reader.yml similarity index 100% rename from dygraph/configs/_base_/readers/mask_fpn_reader.yml rename to dygraph/configs/cascade_rcnn/_base_/cascade_mask_fpn_reader.yml diff --git a/dygraph/configs/_base_/models/cascade_mask_rcnn_r50_fpn.yml b/dygraph/configs/cascade_rcnn/_base_/cascade_mask_rcnn_r50_fpn.yml similarity index 97% rename from dygraph/configs/_base_/models/cascade_mask_rcnn_r50_fpn.yml rename to dygraph/configs/cascade_rcnn/_base_/cascade_mask_rcnn_r50_fpn.yml index 2056eb0ed..7c45b6a6b 100644 --- a/dygraph/configs/_base_/models/cascade_mask_rcnn_r50_fpn.yml +++ b/dygraph/configs/cascade_rcnn/_base_/cascade_mask_rcnn_r50_fpn.yml @@ -1,6 +1,5 @@ architecture: CascadeRCNN pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/ResNet50_cos_pretrained.tar -weights: output/cascade_mask_rcnn_r50_fpn_1x_coco/model_final load_static_weights: True roi_stages: 3 diff --git a/dygraph/configs/_base_/models/cascade_rcnn_r50_fpn.yml b/dygraph/configs/cascade_rcnn/_base_/cascade_rcnn_r50_fpn.yml similarity index 97% rename from dygraph/configs/_base_/models/cascade_rcnn_r50_fpn.yml rename to dygraph/configs/cascade_rcnn/_base_/cascade_rcnn_r50_fpn.yml index 72c337902..26d867c31 100644 --- a/dygraph/configs/_base_/models/cascade_rcnn_r50_fpn.yml +++ b/dygraph/configs/cascade_rcnn/_base_/cascade_rcnn_r50_fpn.yml @@ -1,6 +1,5 @@ architecture: CascadeRCNN pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/ResNet50_cos_pretrained.tar -weights: output/cascade_rcnn_r50_fpn_1x_coco/model_final load_static_weights: True roi_stages: 3 diff --git a/dygraph/configs/_base_/optimizers/rcnn_1x.yml b/dygraph/configs/cascade_rcnn/_base_/optimizer_1x.yml similarity index 100% rename from dygraph/configs/_base_/optimizers/rcnn_1x.yml rename to dygraph/configs/cascade_rcnn/_base_/optimizer_1x.yml diff --git a/dygraph/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco.yml b/dygraph/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco.yml new file mode 100644 index 000000000..b2c7e536d --- /dev/null +++ b/dygraph/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco.yml @@ -0,0 +1,8 @@ +_BASE_: [ + '../datasets/coco_instance.yml', + '../runtime.yml', + '_base_/optimizer_1x.yml', + '_base_/cascade_mask_rcnn_r50_fpn.yml', + '_base_/cascade_mask_fpn_reader.yml', +] +weights: output/cascade_mask_rcnn_r50_fpn_1x_coco/model_final diff --git a/dygraph/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco.yml b/dygraph/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco.yml new file mode 100644 index 000000000..b2cc7993b --- /dev/null +++ b/dygraph/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco.yml @@ -0,0 +1,8 @@ +_BASE_: [ + '../datasets/coco_detection.yml', + '../runtime.yml', + '_base_/optimizer_1x.yml', + '_base_/cascade_rcnn_r50_fpn.yml', + '_base_/cascade_fpn_reader.yml', +] +weights: output/cascade_rcnn_r50_fpn_1x_coco/model_final diff --git a/dygraph/configs/cascade_rcnn_r50_fpn_1x_coco.yml b/dygraph/configs/cascade_rcnn_r50_fpn_1x_coco.yml deleted file mode 100644 index 1f7b471d3..000000000 --- a/dygraph/configs/cascade_rcnn_r50_fpn_1x_coco.yml +++ /dev/null @@ -1,7 +0,0 @@ -_BASE_: [ - './_base_/models/cascade_rcnn_r50_fpn.yml', - './_base_/optimizers/rcnn_1x.yml', - './_base_/datasets/coco_detection.yml', - './_base_/readers/faster_fpn_reader.yml', - './_base_/runtime.yml', -] diff --git a/dygraph/configs/_base_/datasets/coco_detection.yml b/dygraph/configs/datasets/coco_detection.yml similarity index 100% rename from dygraph/configs/_base_/datasets/coco_detection.yml rename to dygraph/configs/datasets/coco_detection.yml diff --git a/dygraph/configs/_base_/datasets/coco_instance.yml b/dygraph/configs/datasets/coco_instance.yml similarity index 100% rename from dygraph/configs/_base_/datasets/coco_instance.yml rename to dygraph/configs/datasets/coco_instance.yml diff --git a/dygraph/configs/_base_/datasets/voc.yml b/dygraph/configs/datasets/voc.yml similarity index 100% rename from dygraph/configs/_base_/datasets/voc.yml rename to dygraph/configs/datasets/voc.yml diff --git a/dygraph/configs/faster_rcnn/README.md b/dygraph/configs/faster_rcnn/README.md new file mode 100644 index 000000000..d4120b325 --- /dev/null +++ b/dygraph/configs/faster_rcnn/README.md @@ -0,0 +1,20 @@ +# Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks + +## Model Zoo + +| 骨架网络 | 网络类型 | 每张GPU图片个数 | 学习率策略 |推理时间(fps) | Box AP | Mask AP | 下载 | 配置文件 | +| :------------------- | :------------- | :-----: | :-----: | :------------: | :-----: | :-----: | :-----------------------------------------------------: | :-----: | +| ResNet50 | Faster | 1 | 1x | ---- | 35.1 | - | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/faster_rcnn_r50_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/faster_rcnn_r50_1x_coco.yml) | +| ResNet50-FPN | Faster | 1 | 1x | ---- | 37.0 | - | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/faster_rcnn_r50_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/faster_rcnn_r50_fpn_1x_coco.yml) | + +## Citations +``` +@article{Ren_2017, + title={Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks}, + journal={IEEE Transactions on Pattern Analysis and Machine Intelligence}, + publisher={Institute of Electrical and Electronics Engineers (IEEE)}, + author={Ren, Shaoqing and He, Kaiming and Girshick, Ross and Sun, Jian}, + year={2017}, + month={Jun}, +} +``` diff --git a/dygraph/configs/faster_rcnn/_base_/faster_fpn_reader.yml b/dygraph/configs/faster_rcnn/_base_/faster_fpn_reader.yml new file mode 100644 index 000000000..8a8fb1ebb --- /dev/null +++ b/dygraph/configs/faster_rcnn/_base_/faster_fpn_reader.yml @@ -0,0 +1,40 @@ +worker_num: 2 +TrainReader: + sample_transforms: + - DecodeOp: { } + - RandomFlipImage: {prob: 0.5} + - NormalizeImage: {is_channel_first: false, is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]} + - ResizeImage: {target_size: 800, max_size: 1333, interp: 1, use_cv2: true} + - Permute: {to_bgr: false, channel_first: true} + batch_transforms: + - PadBatch: {pad_to_stride: 32, use_padded_im_info: false, pad_gt: true} + batch_size: 1 + shuffle: true + drop_last: true + + +EvalReader: + sample_transforms: + - DecodeOp: { } + - NormalizeImageOp: { is_scale: true, mean: [ 0.485,0.456,0.406 ], std: [ 0.229, 0.224,0.225 ] } + - ResizeOp: { interp: 1, target_size: [ 800, 1333 ], keep_ratio: True } + - PermuteOp: { } + batch_transforms: + - PadBatchOp: { pad_to_stride: 32, pad_gt: false } + batch_size: 1 + shuffle: false + drop_last: false + drop_empty: false + + +TestReader: + sample_transforms: + - DecodeOp: { } + - NormalizeImageOp: { is_scale: true, mean: [ 0.485,0.456,0.406 ], std: [ 0.229, 0.224,0.225 ] } + - ResizeOp: { interp: 1, target_size: [ 800, 1333 ], keep_ratio: True } + - PermuteOp: { } + batch_transforms: + - PadBatchOp: { pad_to_stride: 32, pad_gt: false } + batch_size: 1 + shuffle: false + drop_last: false diff --git a/dygraph/configs/_base_/models/faster_rcnn_r50.yml b/dygraph/configs/faster_rcnn/_base_/faster_rcnn_r50.yml similarity index 96% rename from dygraph/configs/_base_/models/faster_rcnn_r50.yml rename to dygraph/configs/faster_rcnn/_base_/faster_rcnn_r50.yml index 9d5b05e87..be0192c62 100644 --- a/dygraph/configs/_base_/models/faster_rcnn_r50.yml +++ b/dygraph/configs/faster_rcnn/_base_/faster_rcnn_r50.yml @@ -1,6 +1,5 @@ architecture: FasterRCNN pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/ResNet50_cos_pretrained.tar -weights: output/faster_rcnn_r50_1x_coco/model_final.pdparams load_static_weights: True # Model Achitecture diff --git a/dygraph/configs/_base_/models/faster_rcnn_r50_fpn.yml b/dygraph/configs/faster_rcnn/_base_/faster_rcnn_r50_fpn.yml similarity index 96% rename from dygraph/configs/_base_/models/faster_rcnn_r50_fpn.yml rename to dygraph/configs/faster_rcnn/_base_/faster_rcnn_r50_fpn.yml index f7f76221c..c831f7f96 100644 --- a/dygraph/configs/_base_/models/faster_rcnn_r50_fpn.yml +++ b/dygraph/configs/faster_rcnn/_base_/faster_rcnn_r50_fpn.yml @@ -1,6 +1,5 @@ architecture: FasterRCNN pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/ResNet50_cos_pretrained.tar -weights: output/faster_rcnn_r50_fpn_1x_coco/model_final.pdparams load_static_weights: True # Model Achitecture diff --git a/dygraph/configs/_base_/readers/faster_reader.yml b/dygraph/configs/faster_rcnn/_base_/faster_reader.yml similarity index 100% rename from dygraph/configs/_base_/readers/faster_reader.yml rename to dygraph/configs/faster_rcnn/_base_/faster_reader.yml diff --git a/dygraph/configs/faster_rcnn/_base_/optimizer_1x.yml b/dygraph/configs/faster_rcnn/_base_/optimizer_1x.yml new file mode 100644 index 000000000..d28b0947b --- /dev/null +++ b/dygraph/configs/faster_rcnn/_base_/optimizer_1x.yml @@ -0,0 +1,19 @@ +epoch: 12 + +LearningRate: + base_lr: 0.01 + schedulers: + - !PiecewiseDecay + gamma: 0.1 + milestones: [8, 11] + - !LinearWarmup + start_factor: 0.3333333333333333 + steps: 500 + +OptimizerBuilder: + optimizer: + momentum: 0.9 + type: Momentum + regularizer: + factor: 0.0001 + type: L2 diff --git a/dygraph/configs/faster_rcnn/faster_rcnn_r50_1x_coco.yml b/dygraph/configs/faster_rcnn/faster_rcnn_r50_1x_coco.yml new file mode 100644 index 000000000..a49bde88a --- /dev/null +++ b/dygraph/configs/faster_rcnn/faster_rcnn_r50_1x_coco.yml @@ -0,0 +1,8 @@ +_BASE_: [ + '../datasets/coco_detection.yml', + '../runtime.yml', + '_base_/optimizer_1x.yml', + '_base_/faster_rcnn_r50.yml', + '_base_/faster_reader.yml', +] +weights: output/faster_rcnn_r50_1x_coco/model_final diff --git a/dygraph/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.yml b/dygraph/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.yml new file mode 100644 index 000000000..e7b451895 --- /dev/null +++ b/dygraph/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.yml @@ -0,0 +1,8 @@ +_BASE_: [ + '../datasets/coco_detection.yml', + '../runtime.yml', + '_base_/optimizer_1x.yml', + '_base_/faster_rcnn_r50_fpn.yml', + '_base_/faster_fpn_reader.yml', +] +weights: output/faster_rcnn_r50_fpn_1x_coco/model_final diff --git a/dygraph/configs/faster_rcnn_r50_1x_coco.yml b/dygraph/configs/faster_rcnn_r50_1x_coco.yml deleted file mode 100644 index f44dc9a87..000000000 --- a/dygraph/configs/faster_rcnn_r50_1x_coco.yml +++ /dev/null @@ -1,7 +0,0 @@ -_BASE_: [ - './_base_/models/faster_rcnn_r50.yml', - './_base_/optimizers/rcnn_1x.yml', - './_base_/datasets/coco_detection.yml', - './_base_/readers/faster_reader.yml', - './_base_/runtime.yml', -] diff --git a/dygraph/configs/faster_rcnn_r50_fpn_1x_coco.yml b/dygraph/configs/faster_rcnn_r50_fpn_1x_coco.yml deleted file mode 100644 index 7ae27da7f..000000000 --- a/dygraph/configs/faster_rcnn_r50_fpn_1x_coco.yml +++ /dev/null @@ -1,7 +0,0 @@ -_BASE_: [ - './_base_/models/faster_rcnn_r50_fpn.yml', - './_base_/optimizers/rcnn_1x.yml', - './_base_/datasets/coco_detection.yml', - './_base_/readers/faster_fpn_reader.yml', - './_base_/runtime.yml', -] diff --git a/dygraph/configs/fcos/fcos_r50_fpn_1x_coco.yml b/dygraph/configs/fcos/fcos_r50_fpn_1x_coco.yml index 7898602d1..0b47d454f 100644 --- a/dygraph/configs/fcos/fcos_r50_fpn_1x_coco.yml +++ b/dygraph/configs/fcos/fcos_r50_fpn_1x_coco.yml @@ -1,6 +1,6 @@ _BASE_: [ - '../_base_/datasets/coco_detection.yml', - '../_base_/runtime.yml', + '../datasets/coco_detection.yml', + '../runtime.yml', '_base_/fcos_r50_fpn.yml', '_base_/optimizer_1x.yml', '_base_/fcos_reader.yml', diff --git a/dygraph/configs/mask_rcnn/README.md b/dygraph/configs/mask_rcnn/README.md new file mode 100644 index 000000000..790ba7e25 --- /dev/null +++ b/dygraph/configs/mask_rcnn/README.md @@ -0,0 +1,20 @@ +# Mask R-CNN + +## Model Zoo + +| 骨架网络 | 网络类型 | 每张GPU图片个数 | 学习率策略 |推理时间(fps) | Box AP | Mask AP | 下载 | 配置文件 | +| :------------------- | :------------- | :-----: | :-----: | :------------: | :-----: | :-----: | :-----------------------------------------------------: | :-----: | +| ResNet50 | Mask | 1 | 1x | ---- | 36.4 | 31.9 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/mask_rcnn_r50_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/mask_rcnn_r50_1x_coco.yml) | +| ResNet50-FPN | Mask | 1 | 1x | ---- | 38.3 | 34.5 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/mask_rcnn_r50_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/mask_rcnn_r50_fpn_1x_coco.yml) | + +## Citations +``` +@article{He_2017, + title={Mask R-CNN}, + journal={2017 IEEE International Conference on Computer Vision (ICCV)}, + publisher={IEEE}, + author={He, Kaiming and Gkioxari, Georgia and Dollar, Piotr and Girshick, Ross}, + year={2017}, + month={Oct} +} +``` diff --git a/dygraph/configs/mask_rcnn/_base_/mask_fpn_reader.yml b/dygraph/configs/mask_rcnn/_base_/mask_fpn_reader.yml new file mode 100644 index 000000000..1d73c7f31 --- /dev/null +++ b/dygraph/configs/mask_rcnn/_base_/mask_fpn_reader.yml @@ -0,0 +1,40 @@ +worker_num: 2 +TrainReader: + sample_transforms: + - DecodeOp: {} + - RandomFlipImage: {prob: 0.5, is_mask_flip: true} + - NormalizeImage: {is_channel_first: false, is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]} + - ResizeImage: {target_size: 800, max_size: 1333, interp: 1, use_cv2: true} + - Permute: {to_bgr: false, channel_first: true} + batch_transforms: + - PadBatch: {pad_to_stride: 32, use_padded_im_info: false, pad_gt: true} + batch_size: 1 + shuffle: true + drop_last: true + + +EvalReader: + sample_transforms: + - DecodeOp: {} + - NormalizeImageOp: {is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]} + - ResizeOp: {interp: 1, target_size: [800, 1333], keep_ratio: True} + - PermuteOp: {} + batch_transforms: + - PadBatchOp: {pad_to_stride: 32, pad_gt: false} + batch_size: 1 + shuffle: false + drop_last: false + drop_empty: false + + +TestReader: + sample_transforms: + - DecodeOp: {} + - NormalizeImageOp: {is_scale: true, mean: [0.485,0.456,0.406], std: [0.229, 0.224,0.225]} + - ResizeOp: {interp: 1, target_size: [800, 1333], keep_ratio: True} + - PermuteOp: {} + batch_transforms: + - PadBatchOp: {pad_to_stride: 32, pad_gt: false} + batch_size: 1 + shuffle: false + drop_last: false diff --git a/dygraph/configs/_base_/models/mask_rcnn_r50.yml b/dygraph/configs/mask_rcnn/_base_/mask_rcnn_r50.yml similarity index 97% rename from dygraph/configs/_base_/models/mask_rcnn_r50.yml rename to dygraph/configs/mask_rcnn/_base_/mask_rcnn_r50.yml index ffc8dab00..eac16e105 100644 --- a/dygraph/configs/_base_/models/mask_rcnn_r50.yml +++ b/dygraph/configs/mask_rcnn/_base_/mask_rcnn_r50.yml @@ -1,6 +1,5 @@ architecture: MaskRCNN pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/ResNet50_cos_pretrained.tar -weights: output/mask_rcnn_r50_fpn_1x/model_final load_static_weights: True # Model Achitecture diff --git a/dygraph/configs/_base_/models/mask_rcnn_r50_fpn.yml b/dygraph/configs/mask_rcnn/_base_/mask_rcnn_r50_fpn.yml similarity index 97% rename from dygraph/configs/_base_/models/mask_rcnn_r50_fpn.yml rename to dygraph/configs/mask_rcnn/_base_/mask_rcnn_r50_fpn.yml index e0464497d..403a0b43d 100644 --- a/dygraph/configs/_base_/models/mask_rcnn_r50_fpn.yml +++ b/dygraph/configs/mask_rcnn/_base_/mask_rcnn_r50_fpn.yml @@ -1,6 +1,5 @@ architecture: MaskRCNN pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/ResNet50_cos_pretrained.tar -weights: output/mask_rcnn_r50_fpn_1x/model_final load_static_weights: True # Model Achitecture diff --git a/dygraph/configs/_base_/readers/mask_reader.yml b/dygraph/configs/mask_rcnn/_base_/mask_reader.yml similarity index 100% rename from dygraph/configs/_base_/readers/mask_reader.yml rename to dygraph/configs/mask_rcnn/_base_/mask_reader.yml diff --git a/dygraph/configs/mask_rcnn/_base_/optimizer_1x.yml b/dygraph/configs/mask_rcnn/_base_/optimizer_1x.yml new file mode 100644 index 000000000..d28b0947b --- /dev/null +++ b/dygraph/configs/mask_rcnn/_base_/optimizer_1x.yml @@ -0,0 +1,19 @@ +epoch: 12 + +LearningRate: + base_lr: 0.01 + schedulers: + - !PiecewiseDecay + gamma: 0.1 + milestones: [8, 11] + - !LinearWarmup + start_factor: 0.3333333333333333 + steps: 500 + +OptimizerBuilder: + optimizer: + momentum: 0.9 + type: Momentum + regularizer: + factor: 0.0001 + type: L2 diff --git a/dygraph/configs/mask_rcnn/mask_rcnn_r50_1x_coco.yml b/dygraph/configs/mask_rcnn/mask_rcnn_r50_1x_coco.yml new file mode 100644 index 000000000..01f4721cb --- /dev/null +++ b/dygraph/configs/mask_rcnn/mask_rcnn_r50_1x_coco.yml @@ -0,0 +1,8 @@ +_BASE_: [ + '../datasets/coco_instance.yml', + '../runtime.yml', + '_base_/optimizer_1x.yml', + '_base_/mask_rcnn_r50.yml', + '_base_/mask_reader.yml', +] +weights: output/mask_rcnn_r50_1x_coco/model_final diff --git a/dygraph/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco.yml b/dygraph/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco.yml new file mode 100644 index 000000000..95e48c2b4 --- /dev/null +++ b/dygraph/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco.yml @@ -0,0 +1,8 @@ +_BASE_: [ + '../datasets/coco_instance.yml', + '../runtime.yml', + '_base_/optimizer_1x.yml', + '_base_/mask_rcnn_r50_fpn.yml', + '_base_/mask_fpn_reader.yml', +] +weights: output/mask_rcnn_r50_fpn_1x_coco/model_final diff --git a/dygraph/configs/mask_rcnn_r50_1x_coco.yml b/dygraph/configs/mask_rcnn_r50_1x_coco.yml deleted file mode 100644 index 50e48272a..000000000 --- a/dygraph/configs/mask_rcnn_r50_1x_coco.yml +++ /dev/null @@ -1,7 +0,0 @@ -_BASE_: [ - './_base_/models/mask_rcnn_r50.yml', - './_base_/optimizers/rcnn_1x.yml', - './_base_/datasets/coco_instance.yml', - './_base_/readers/mask_reader.yml', - './_base_/runtime.yml', -] diff --git a/dygraph/configs/mask_rcnn_r50_fpn_1x_coco.yml b/dygraph/configs/mask_rcnn_r50_fpn_1x_coco.yml deleted file mode 100644 index b881e14cd..000000000 --- a/dygraph/configs/mask_rcnn_r50_fpn_1x_coco.yml +++ /dev/null @@ -1,7 +0,0 @@ -_BASE_: [ - './_base_/models/mask_rcnn_r50_fpn.yml', - './_base_/optimizers/rcnn_1x.yml', - './_base_/datasets/coco_instance.yml', - './_base_/readers/mask_fpn_reader.yml', - './_base_/runtime.yml', -] diff --git a/dygraph/configs/_base_/runtime.yml b/dygraph/configs/runtime.yml similarity index 100% rename from dygraph/configs/_base_/runtime.yml rename to dygraph/configs/runtime.yml diff --git a/dygraph/configs/ssd/README.md b/dygraph/configs/ssd/README.md new file mode 100644 index 000000000..e8593eac1 --- /dev/null +++ b/dygraph/configs/ssd/README.md @@ -0,0 +1,21 @@ +# SSD: Single Shot MultiBox Detector + +## Model Zoo + +### SSD on Pascal VOC + +| 骨架网络 | 网络类型 | 每张GPU图片个数 | 学习率策略 |推理时间(fps) | Box AP | 下载 | 配置文件 | +| :-------------- | :------------- | :-----: | :-----: | :------------: | :-----: | :-----------------------------------------------------: | :-----: | +| VGG | SSD | 8 | 240e | ---- | 78.2 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/ssd_vgg16_300_240e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/ssd_vgg16_300_240e_voc.yml) | + +**注意:** SSD使用4GPU训练,训练240个epoch + +## Citations +``` +@article{Liu_2016, + title={SSD: Single Shot MultiBox Detector}, + journal={ECCV}, + author={Liu, Wei and Anguelov, Dragomir and Erhan, Dumitru and Szegedy, Christian and Reed, Scott and Fu, Cheng-Yang and Berg, Alexander C.}, + year={2016}, +} +``` diff --git a/dygraph/configs/_base_/optimizers/ssdlite_1000e.yml b/dygraph/configs/ssd/_base_/optimizer_1000e.yml similarity index 100% rename from dygraph/configs/_base_/optimizers/ssdlite_1000e.yml rename to dygraph/configs/ssd/_base_/optimizer_1000e.yml diff --git a/dygraph/configs/_base_/optimizers/ssd_mobilenet_120e.yml b/dygraph/configs/ssd/_base_/optimizer_120e.yml similarity index 100% rename from dygraph/configs/_base_/optimizers/ssd_mobilenet_120e.yml rename to dygraph/configs/ssd/_base_/optimizer_120e.yml diff --git a/dygraph/configs/_base_/optimizers/ssd_240e.yml b/dygraph/configs/ssd/_base_/optimizer_240e.yml similarity index 100% rename from dygraph/configs/_base_/optimizers/ssd_240e.yml rename to dygraph/configs/ssd/_base_/optimizer_240e.yml diff --git a/dygraph/configs/_base_/readers/ssd_mobilenet_reader.yml b/dygraph/configs/ssd/_base_/ssd_mobilenet_reader.yml similarity index 100% rename from dygraph/configs/_base_/readers/ssd_mobilenet_reader.yml rename to dygraph/configs/ssd/_base_/ssd_mobilenet_reader.yml diff --git a/dygraph/configs/_base_/models/ssd_mobilenet_v1_300.yml b/dygraph/configs/ssd/_base_/ssd_mobilenet_v1_300.yml similarity index 96% rename from dygraph/configs/_base_/models/ssd_mobilenet_v1_300.yml rename to dygraph/configs/ssd/_base_/ssd_mobilenet_v1_300.yml index f061ec4c4..c277a5dfc 100644 --- a/dygraph/configs/_base_/models/ssd_mobilenet_v1_300.yml +++ b/dygraph/configs/ssd/_base_/ssd_mobilenet_v1_300.yml @@ -1,6 +1,6 @@ architecture: SSD pretrain_weights: https://paddlemodels.bj.bcebos.com/object_detection/ssd_mobilenet_v1_coco_pretrained.tar -weights: output/ssd_mobilenetv1/model_final +load_static_weights: True SSD: backbone: MobileNet diff --git a/dygraph/configs/_base_/readers/ssd_reader.yml b/dygraph/configs/ssd/_base_/ssd_reader.yml similarity index 100% rename from dygraph/configs/_base_/readers/ssd_reader.yml rename to dygraph/configs/ssd/_base_/ssd_reader.yml diff --git a/dygraph/configs/_base_/models/ssd_vgg16_300.yml b/dygraph/configs/ssd/_base_/ssd_vgg16_300.yml similarity index 96% rename from dygraph/configs/_base_/models/ssd_vgg16_300.yml rename to dygraph/configs/ssd/_base_/ssd_vgg16_300.yml index 497292b5a..63ebc4ae0 100644 --- a/dygraph/configs/_base_/models/ssd_vgg16_300.yml +++ b/dygraph/configs/ssd/_base_/ssd_vgg16_300.yml @@ -1,6 +1,6 @@ architecture: SSD pretrain_weights: https://paddlemodels.bj.bcebos.com/object_detection/dygraph/VGG16_caffe_pretrained.pdparams -weights: output/ssd_vgg16/model_final +load_static_weights: True # Model Achitecture SSD: diff --git a/dygraph/configs/_base_/readers/ssdlite300_reader.yml b/dygraph/configs/ssd/_base_/ssdlite300_reader.yml similarity index 100% rename from dygraph/configs/_base_/readers/ssdlite300_reader.yml rename to dygraph/configs/ssd/_base_/ssdlite300_reader.yml diff --git a/dygraph/configs/_base_/readers/ssdlite320_reader.yml b/dygraph/configs/ssd/_base_/ssdlite320_reader.yml similarity index 100% rename from dygraph/configs/_base_/readers/ssdlite320_reader.yml rename to dygraph/configs/ssd/_base_/ssdlite320_reader.yml diff --git a/dygraph/configs/_base_/models/ssdlite_mobilenet_v1_300.yml b/dygraph/configs/ssd/_base_/ssdlite_mobilenet_v1_300.yml similarity index 95% rename from dygraph/configs/_base_/models/ssdlite_mobilenet_v1_300.yml rename to dygraph/configs/ssd/_base_/ssdlite_mobilenet_v1_300.yml index 1b8a10885..95847a0b9 100644 --- a/dygraph/configs/_base_/models/ssdlite_mobilenet_v1_300.yml +++ b/dygraph/configs/ssd/_base_/ssdlite_mobilenet_v1_300.yml @@ -1,6 +1,6 @@ architecture: SSD pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/MobileNetV1_ssld_pretrained.tar -weights: output/ssdlite_mobilenet_v1/model_final +load_static_weights: True SSD: backbone: MobileNet diff --git a/dygraph/configs/_base_/models/ssdlite_mobilenet_v3_large_320.yml b/dygraph/configs/ssd/_base_/ssdlite_mobilenet_v3_large_320.yml similarity index 95% rename from dygraph/configs/_base_/models/ssdlite_mobilenet_v3_large_320.yml rename to dygraph/configs/ssd/_base_/ssdlite_mobilenet_v3_large_320.yml index bc93f9de1..e7c2b1c3d 100644 --- a/dygraph/configs/_base_/models/ssdlite_mobilenet_v3_large_320.yml +++ b/dygraph/configs/ssd/_base_/ssdlite_mobilenet_v3_large_320.yml @@ -1,6 +1,6 @@ architecture: SSD pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/MobileNetV3_large_x1_0_ssld_pretrained.tar -weights: output/ssdlite_mobilenet_v3_large/model_final +load_static_weights: True SSD: backbone: MobileNetV3 diff --git a/dygraph/configs/_base_/models/ssdlite_mobilenet_v3_small_320.yml b/dygraph/configs/ssd/_base_/ssdlite_mobilenet_v3_small_320.yml similarity index 95% rename from dygraph/configs/_base_/models/ssdlite_mobilenet_v3_small_320.yml rename to dygraph/configs/ssd/_base_/ssdlite_mobilenet_v3_small_320.yml index 3d7a789fa..21dfe669a 100644 --- a/dygraph/configs/_base_/models/ssdlite_mobilenet_v3_small_320.yml +++ b/dygraph/configs/ssd/_base_/ssdlite_mobilenet_v3_small_320.yml @@ -1,6 +1,6 @@ architecture: SSD pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/MobileNetV3_small_x1_0_ssld_pretrained.tar -weights: output/ssd_mobilenet_v3_small/model_final +load_static_weights: True SSD: backbone: MobileNetV3 diff --git a/dygraph/configs/ssd/ssd_mobilenet_v1_300_120e_voc.yml b/dygraph/configs/ssd/ssd_mobilenet_v1_300_120e_voc.yml new file mode 100644 index 000000000..45de7733a --- /dev/null +++ b/dygraph/configs/ssd/ssd_mobilenet_v1_300_120e_voc.yml @@ -0,0 +1,8 @@ +_BASE_: [ + '../datasets/voc.yml', + '../runtime.yml', + '_base_/optimizer_120e.yml', + '_base_/ssd_mobilenet_v1_300.yml', + '_base_/ssd_mobilenet_reader.yml', +] +weights: output/ssd_mobilenet_v1_300_120e_voc/model_final diff --git a/dygraph/configs/ssd/ssd_vgg16_300_240e_voc.yml b/dygraph/configs/ssd/ssd_vgg16_300_240e_voc.yml new file mode 100644 index 000000000..58cf4b985 --- /dev/null +++ b/dygraph/configs/ssd/ssd_vgg16_300_240e_voc.yml @@ -0,0 +1,8 @@ +_BASE_: [ + '../datasets/voc.yml', + '../runtime.yml', + '_base_/optimizer_240e.yml', + '_base_/ssd_vgg16_300.yml', + '_base_/ssd_reader.yml', +] +weights: output/ssd_vgg16_300_240e_voc/model_final diff --git a/dygraph/configs/ssd/ssdlite_mobilenet_v1_300_coco.yml b/dygraph/configs/ssd/ssdlite_mobilenet_v1_300_coco.yml new file mode 100644 index 000000000..d3a649adf --- /dev/null +++ b/dygraph/configs/ssd/ssdlite_mobilenet_v1_300_coco.yml @@ -0,0 +1,8 @@ +_BASE_: [ + '../datasets/coco_detection.yml', + '../runtime.yml', + '_base_/optimizer_1000e.yml', + '_base_/ssdlite_mobilenet_v1_300.yml', + '_base_/ssdlite300_reader.yml', +] +weights: output/ssdlite_mobilenet_v1_300_coco/model_final diff --git a/dygraph/configs/ssd/ssdlite_mobilenet_v3_large_320_coco.yml b/dygraph/configs/ssd/ssdlite_mobilenet_v3_large_320_coco.yml new file mode 100644 index 000000000..54fb553a0 --- /dev/null +++ b/dygraph/configs/ssd/ssdlite_mobilenet_v3_large_320_coco.yml @@ -0,0 +1,8 @@ +_BASE_: [ + '../datasets/coco_detection.yml', + '../runtime.yml', + '_base_/optimizer_1000e.yml', + '_base_/ssdlite_mobilenet_v3_large_320.yml', + '_base_/ssdlite320_reader.yml', +] +weights: output/ssdlite_mobilenet_v3_large_320_coco/model_final diff --git a/dygraph/configs/ssd/ssdlite_mobilenet_v3_small_320_coco.yml b/dygraph/configs/ssd/ssdlite_mobilenet_v3_small_320_coco.yml new file mode 100644 index 000000000..1b696e7b9 --- /dev/null +++ b/dygraph/configs/ssd/ssdlite_mobilenet_v3_small_320_coco.yml @@ -0,0 +1,8 @@ +_BASE_: [ + '../datasets/coco_detection.yml', + '../runtime.yml', + '_base_/optimizer_1000e.yml', + '_base_/ssdlite_mobilenet_v3_small_320.yml', + '_base_/ssdlite320_reader.yml', +] +weights: output/ssdlite_mobilenet_v3_small_320_coco/model_final diff --git a/dygraph/configs/ssd_mobilenet_v1_300_120e_voc.yml b/dygraph/configs/ssd_mobilenet_v1_300_120e_voc.yml deleted file mode 100644 index deb942f38..000000000 --- a/dygraph/configs/ssd_mobilenet_v1_300_120e_voc.yml +++ /dev/null @@ -1,7 +0,0 @@ -_BASE_: [ - './_base_/models/ssd_mobilenet_v1_300.yml', - './_base_/optimizers/ssd_mobilenet_120e.yml', - './_base_/datasets/voc.yml', - './_base_/readers/ssd_mobilenet_reader.yml', - './_base_/runtime.yml', -] diff --git a/dygraph/configs/ssd_vgg16_300_240e_voc.yml b/dygraph/configs/ssd_vgg16_300_240e_voc.yml deleted file mode 100644 index 0052a3e03..000000000 --- a/dygraph/configs/ssd_vgg16_300_240e_voc.yml +++ /dev/null @@ -1,7 +0,0 @@ -_BASE_: [ - './_base_/models/ssd_vgg16_300.yml', - './_base_/optimizers/ssd_240e.yml', - './_base_/datasets/voc.yml', - './_base_/readers/ssd_reader.yml', - './_base_/runtime.yml', -] diff --git a/dygraph/configs/ssdlite_mobilenet_v1_300_coco.yml b/dygraph/configs/ssdlite_mobilenet_v1_300_coco.yml deleted file mode 100644 index afd3dab09..000000000 --- a/dygraph/configs/ssdlite_mobilenet_v1_300_coco.yml +++ /dev/null @@ -1,7 +0,0 @@ -_BASE_: [ - './_base_/models/ssdlite_mobilenet_v1_300.yml', - './_base_/optimizers/ssdlite_1000e.yml', - './_base_/datasets/coco_detection.yml', - './_base_/readers/ssdlite300_reader.yml', - './_base_/runtime.yml', -] diff --git a/dygraph/configs/ssdlite_mobilenet_v3_large_320_coco.yml b/dygraph/configs/ssdlite_mobilenet_v3_large_320_coco.yml deleted file mode 100644 index 3be024e5e..000000000 --- a/dygraph/configs/ssdlite_mobilenet_v3_large_320_coco.yml +++ /dev/null @@ -1,7 +0,0 @@ -_BASE_: [ - './_base_/models/ssdlite_mobilenet_v3_large_320.yml', - './_base_/optimizers/ssdlite_1000e.yml', - './_base_/datasets/coco_detection.yml', - './_base_/readers/ssdlite320_reader.yml', - './_base_/runtime.yml', -] diff --git a/dygraph/configs/ssdlite_mobilenet_v3_small_320_coco.yml b/dygraph/configs/ssdlite_mobilenet_v3_small_320_coco.yml deleted file mode 100644 index 0097beb96..000000000 --- a/dygraph/configs/ssdlite_mobilenet_v3_small_320_coco.yml +++ /dev/null @@ -1,7 +0,0 @@ -_BASE_: [ - './_base_/models/ssdlite_mobilenet_v3_small_320.yml', - './_base_/optimizers/ssdlite_1000e.yml', - './_base_/datasets/coco_detection.yml', - './_base_/readers/ssdlite320_reader.yml', - './_base_/runtime.yml', -] diff --git a/dygraph/configs/yolov3/README.md b/dygraph/configs/yolov3/README.md new file mode 100644 index 000000000..98cc49b79 --- /dev/null +++ b/dygraph/configs/yolov3/README.md @@ -0,0 +1,46 @@ +# YOLOv3 + +## Model Zoo + +### YOLOv3 on COCO + +| 骨架网络 | 输入尺寸 | 每张GPU图片个数 | 学习率策略 |推理时间(fps) | Box AP | 下载 | 配置文件 | +| :------------------- | :------- | :-----: | :-----: | :------------: | :-----: | :-----------------------------------------------------: | :-----: | +| DarkNet53(paper) | 608 | 8 | 270e | ---- | 33.0 | - | - | +| DarkNet53(paper) | 416 | 8 | 270e | ---- | 31.0 | - | - | +| DarkNet53(paper) | 320 | 8 | 270e | ---- | 28.2 | - | - | +| DarkNet53 | 608 | 8 | 270e | ---- | 39.0 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_darknet53_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_darknet53_270e_coco.yml) | +| DarkNet53 | 416 | 8 | 270e | ---- | 37.5 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_darknet53_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_darknet53_270e_coco.yml) | +| DarkNet53 | 320 | 8 | 270e | ---- | 34.6 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_darknet53_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_darknet53_270e_coco.yml) | +| MobileNet-V1 | 608 | 8 | 270e | ---- | 28.8 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v1_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v1_270e_coco.yml) | +| MobileNet-V1 | 416 | 8 | 270e | ---- | 28.7 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v1_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v1_270e_coco.yml) | +| MobileNet-V1 | 320 | 8 | 270e | ---- | 26.5 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v1_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v1_270e_coco.yml) | +| MobileNet-V3 | 608 | 8 | 270e | ---- | 31.4 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v3_large_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v3_large_270e_coco.yml) | +| MobileNet-V3 | 416 | 8 | 270e | ---- | 29.7 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v3_large_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v3_large_270e_coco.yml) | +| MobileNet-V3 | 320 | 8 | 270e | ---- | 26.9 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v3_large_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v3_large_270e_coco.yml) | + +### YOLOv3 on Pasacl VOC + +| 骨架网络 | 输入尺寸 | 每张GPU图片个数 | 学习率策略 |推理时间(fps)| Box AP | 下载 | 配置文件 | +| :----------- | :--: | :-----: | :-----: |:------------: |:----: | :-------: | :----: | +| MobileNet-V1 | 608 | 8 | 270e | - | 75.1 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v1_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v1_270e_voc.yml) | +| MobileNet-V1 | 416 | 8 | 270e | - | 76.1 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v1_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v1_270e_voc.yml) | +| MobileNet-V1 | 320 | 8 | 270e | - | 73.6 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v1_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v1_270e_voc.yml) | +| MobileNet-V3 | 608 | 8 | 270e | - | 79.6 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v3_large_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v3_large_270e_voc.yml) | +| MobileNet-V3 | 416 | 8 | 270e | - | 78.6 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v3_large_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v3_large_270e_voc.yml) | +| MobileNet-V3 | 320 | 8 | 270e | - | 76.4 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v3_large_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v3_large_270e_voc.yml) | + +**注意:** YOLOv3均使用8GPU训练,训练270个epoch + + +## Citations +``` +@misc{redmon2018yolov3, + title={YOLOv3: An Incremental Improvement}, + author={Joseph Redmon and Ali Farhadi}, + year={2018}, + eprint={1804.02767}, + archivePrefix={arXiv}, + primaryClass={cs.CV} +} +``` diff --git a/dygraph/configs/_base_/optimizers/yolov3_270e.yml b/dygraph/configs/yolov3/_base_/optimizer_270e.yml similarity index 100% rename from dygraph/configs/_base_/optimizers/yolov3_270e.yml rename to dygraph/configs/yolov3/_base_/optimizer_270e.yml diff --git a/dygraph/configs/_base_/models/yolov3_darknet53.yml b/dygraph/configs/yolov3/_base_/yolov3_darknet53.yml similarity index 95% rename from dygraph/configs/_base_/models/yolov3_darknet53.yml rename to dygraph/configs/yolov3/_base_/yolov3_darknet53.yml index 42efdfdb9..c5615fb0a 100644 --- a/dygraph/configs/_base_/models/yolov3_darknet53.yml +++ b/dygraph/configs/yolov3/_base_/yolov3_darknet53.yml @@ -1,6 +1,5 @@ architecture: YOLOv3 pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/DarkNet53_pretrained.tar -weights: output/yolov3_darknet/model_final use_fine_grained_loss: false load_static_weights: True norm_type: sync_bn diff --git a/dygraph/configs/_base_/models/yolov3_mobilenet_v1.yml b/dygraph/configs/yolov3/_base_/yolov3_mobilenet_v1.yml similarity index 95% rename from dygraph/configs/_base_/models/yolov3_mobilenet_v1.yml rename to dygraph/configs/yolov3/_base_/yolov3_mobilenet_v1.yml index cb39b6119..226e35c50 100644 --- a/dygraph/configs/_base_/models/yolov3_mobilenet_v1.yml +++ b/dygraph/configs/yolov3/_base_/yolov3_mobilenet_v1.yml @@ -1,6 +1,5 @@ architecture: YOLOv3 pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/MobileNetV1_pretrained.tar -weights: output/yolov3_mobilenet_v1/model_final load_static_weights: True norm_type: sync_bn diff --git a/dygraph/configs/_base_/models/yolov3_mobilenet_v3_large.yml b/dygraph/configs/yolov3/_base_/yolov3_mobilenet_v3_large.yml similarity index 95% rename from dygraph/configs/_base_/models/yolov3_mobilenet_v3_large.yml rename to dygraph/configs/yolov3/_base_/yolov3_mobilenet_v3_large.yml index 1b93cb452..0dc8188c4 100644 --- a/dygraph/configs/_base_/models/yolov3_mobilenet_v3_large.yml +++ b/dygraph/configs/yolov3/_base_/yolov3_mobilenet_v3_large.yml @@ -1,6 +1,5 @@ architecture: YOLOv3 pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/MobileNetV3_large_x1_0_pretrained.tar -weights: output/yolov3_mobilenet_v3_large/model_final load_static_weights: True norm_type: sync_bn diff --git a/dygraph/configs/_base_/models/yolov3_mobilenet_v3_small.yml b/dygraph/configs/yolov3/_base_/yolov3_mobilenet_v3_small.yml similarity index 95% rename from dygraph/configs/_base_/models/yolov3_mobilenet_v3_small.yml rename to dygraph/configs/yolov3/_base_/yolov3_mobilenet_v3_small.yml index 53a6819fa..87dd0a7a0 100644 --- a/dygraph/configs/_base_/models/yolov3_mobilenet_v3_small.yml +++ b/dygraph/configs/yolov3/_base_/yolov3_mobilenet_v3_small.yml @@ -1,6 +1,5 @@ architecture: YOLOv3 pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/MobileNetV3_small_x1_0_pretrained.tar -weights: output/yolov3_mobilenet_v3_small/model_final load_static_weights: True norm_type: sync_bn diff --git a/dygraph/configs/_base_/readers/yolov3_reader.yml b/dygraph/configs/yolov3/_base_/yolov3_reader.yml similarity index 100% rename from dygraph/configs/_base_/readers/yolov3_reader.yml rename to dygraph/configs/yolov3/_base_/yolov3_reader.yml diff --git a/dygraph/configs/yolov3/yolov3_darknet53_270e_coco.yml b/dygraph/configs/yolov3/yolov3_darknet53_270e_coco.yml new file mode 100644 index 000000000..15bc38e6c --- /dev/null +++ b/dygraph/configs/yolov3/yolov3_darknet53_270e_coco.yml @@ -0,0 +1,8 @@ +_BASE_: [ + '../datasets/coco_detection.yml', + '../runtime.yml', + '_base_/optimizer_270e.yml', + '_base_/yolov3_darknet53.yml', + '_base_/yolov3_reader.yml', +] +weights: output/yolov3_darknet53_coco/model_final diff --git a/dygraph/configs/yolov3/yolov3_mobilenet_v1_270e_coco.yml b/dygraph/configs/yolov3/yolov3_mobilenet_v1_270e_coco.yml new file mode 100644 index 000000000..4c3c30427 --- /dev/null +++ b/dygraph/configs/yolov3/yolov3_mobilenet_v1_270e_coco.yml @@ -0,0 +1,8 @@ +_BASE_: [ + '../datasets/coco_detection.yml', + '../runtime.yml', + '_base_/optimizer_270e.yml', + '_base_/yolov3_mobilenet_v1.yml', + '_base_/yolov3_reader.yml', +] +weights: output/yolov3_mobilenet_v1_coco/model_final diff --git a/dygraph/configs/yolov3_mobilenet_v1_270e_voc.yml b/dygraph/configs/yolov3/yolov3_mobilenet_v1_270e_voc.yml similarity index 85% rename from dygraph/configs/yolov3_mobilenet_v1_270e_voc.yml rename to dygraph/configs/yolov3/yolov3_mobilenet_v1_270e_voc.yml index 8b54d7209..5d4fb1929 100644 --- a/dygraph/configs/yolov3_mobilenet_v1_270e_voc.yml +++ b/dygraph/configs/yolov3/yolov3_mobilenet_v1_270e_voc.yml @@ -1,10 +1,11 @@ _BASE_: [ - './_base_/models/yolov3_mobilenet_v1.yml', - './_base_/optimizers/yolov3_270e.yml', - './_base_/datasets/voc.yml', - './_base_/readers/yolov3_reader.yml', - './_base_/runtime.yml', + '../datasets/voc.yml', + '../runtime.yml', + '_base_/optimizer_270e.yml', + '_base_/yolov3_mobilenet_v1.yml', + '_base_/yolov3_reader.yml', ] +weights: output/yolov3_mobilenet_v1_voc/model_final TrainReader: inputs_def: diff --git a/dygraph/configs/yolov3/yolov3_mobilenet_v3_large_270e_coco.yml b/dygraph/configs/yolov3/yolov3_mobilenet_v3_large_270e_coco.yml new file mode 100644 index 000000000..dd06756ac --- /dev/null +++ b/dygraph/configs/yolov3/yolov3_mobilenet_v3_large_270e_coco.yml @@ -0,0 +1,8 @@ +_BASE_: [ + '../datasets/coco_detection.yml', + '../runtime.yml', + '_base_/optimizer_270e.yml', + '_base_/yolov3_mobilenet_v3_large.yml', + '_base_/yolov3_reader.yml', +] +weights: output/yolov3_mobilenet_v3_large_coco/model_final diff --git a/dygraph/configs/yolov3_mobilenet_v3_large_270e_voc.yml b/dygraph/configs/yolov3/yolov3_mobilenet_v3_large_270e_voc.yml similarity index 85% rename from dygraph/configs/yolov3_mobilenet_v3_large_270e_voc.yml rename to dygraph/configs/yolov3/yolov3_mobilenet_v3_large_270e_voc.yml index 51495933e..26cd965d3 100644 --- a/dygraph/configs/yolov3_mobilenet_v3_large_270e_voc.yml +++ b/dygraph/configs/yolov3/yolov3_mobilenet_v3_large_270e_voc.yml @@ -1,10 +1,11 @@ _BASE_: [ - './_base_/models/yolov3_mobilenet_v3_large.yml', - './_base_/optimizers/yolov3_270e.yml', - './_base_/datasets/voc.yml', - './_base_/readers/yolov3_reader.yml', - './_base_/runtime.yml', + '../datasets/voc.yml', + '../runtime.yml', + '_base_/optimizer_270e.yml', + '_base_/yolov3_mobilenet_v3_large.yml', + '_base_/yolov3_reader.yml', ] +weights: output/yolov3_mobilenet_v3_large_voc/model_final TrainReader: inputs_def: diff --git a/dygraph/configs/yolov3_darknet53_270e_coco.yml b/dygraph/configs/yolov3_darknet53_270e_coco.yml deleted file mode 100644 index 08ea8bad7..000000000 --- a/dygraph/configs/yolov3_darknet53_270e_coco.yml +++ /dev/null @@ -1,7 +0,0 @@ -_BASE_: [ - './_base_/models/yolov3_darknet53.yml', - './_base_/optimizers/yolov3_270e.yml', - './_base_/datasets/coco_detection.yml', - './_base_/readers/yolov3_reader.yml', - './_base_/runtime.yml', -] diff --git a/dygraph/configs/yolov3_mobilenet_v1_270e_coco.yml b/dygraph/configs/yolov3_mobilenet_v1_270e_coco.yml deleted file mode 100644 index cf82cc02b..000000000 --- a/dygraph/configs/yolov3_mobilenet_v1_270e_coco.yml +++ /dev/null @@ -1,7 +0,0 @@ -_BASE_: [ - './_base_/models/yolov3_mobilenet_v1.yml', - './_base_/optimizers/yolov3_270e.yml', - './_base_/datasets/coco_detection.yml', - './_base_/readers/yolov3_reader.yml', - './_base_/runtime.yml', -] diff --git a/dygraph/configs/yolov3_mobilenet_v3_large_270e_coco.yml b/dygraph/configs/yolov3_mobilenet_v3_large_270e_coco.yml deleted file mode 100644 index fecea2b16..000000000 --- a/dygraph/configs/yolov3_mobilenet_v3_large_270e_coco.yml +++ /dev/null @@ -1,7 +0,0 @@ -_BASE_: [ - './_base_/models/yolov3_mobilenet_v3_large.yml', - './_base_/optimizers/yolov3_270e.yml', - './_base_/datasets/coco_detection.yml', - './_base_/readers/yolov3_reader.yml', - './_base_/runtime.yml', -] diff --git a/dygraph/docs/MODEL_ZOO_cn.md b/dygraph/docs/MODEL_ZOO_cn.md index 66d08e0db..43dd619c8 100644 --- a/dygraph/docs/MODEL_ZOO_cn.md +++ b/dygraph/docs/MODEL_ZOO_cn.md @@ -23,67 +23,35 @@ ## ImageNet预训练模型 -Paddle提供基于ImageNet的骨架网络预训练模型。所有预训练模型均通过标准的Imagenet-1k数据集训练得到。[下载链接](https://github.com/PaddlePaddle/models/tree/develop/PaddleCV/image_classification#supported-models-and-performances) +Paddle提供基于ImageNet的骨架网络预训练模型。所有预训练模型均通过标准的Imagenet-1k数据集训练得到,ResNet和MobileNet等是采用余弦学习率调整策略或SSLD知识蒸馏训练得到的高精度预训练模型,可在[PaddleClas](https://github.com/PaddlePaddle/PaddleClas)查看模型细节。 -- 注:ResNet50模型通过余弦学习率调整策略训练得到。[ResNet50下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/ResNet18_pretrained.tar), - [ResNet50_vd下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/ResNet50_vd_pretrained.tar) ## 基线 -### Faster & Mask R-CNN +### Faster R-CNN -| 骨架网络 | 网络类型 | 每张GPU图片个数 | 学习率策略 |推理时间(fps) | Box AP | Mask AP | 下载 | 配置文件 | -| :------------------- | :------------- | :-----: | :-----: | :------------: | :-----: | :-----: | :-----------------------------------------------------: | :-----: | -| ResNet50 | Faster | 1 | 1x | ---- | 35.1 | - | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/faster_rcnn_r50_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/faster_rcnn_r50_1x_coco.yml) | -| ResNet50-FPN | Faster | 1 | 1x | ---- | 37.0 | - | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/faster_rcnn_r50_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/faster_rcnn_r50_fpn_1x_coco.yml) | -| ResNet50 | Mask | 1 | 1x | ---- | 36.4 | 31.9 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/mask_rcnn_r50_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/mask_rcnn_r50_1x_coco.yml) | -| ResNet50-FPN | Mask | 1 | 1x | ---- | 38.3 | 34.5 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/mask_rcnn_r50_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/mask_rcnn_r50_fpn_1x_coco.yml) | -| ResNet50-FPN | Cascade Faster | 1 | 1x | ---- | 41.1 | - | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/cascade_rcnn_r50_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/cascade_faster_rcnn_r50_fpn_1x_coco.yml) | -| ResNet50-FPN | Cascade Mask | 1 | 1x | ---- | 41.6 | 35.3 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/cascade_mask_rcnn_r50_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/cascade_mask_rcnn_r50_fpn_1x_coco.yml) | +请参考[Faster R-CNN](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/faster_rcnn/) +### Mask R-CNN -### YOLOv3 on COCO +请参考[Mask R-CNN](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/mask_rcnn/) -| 骨架网络 | 输入尺寸 | 每张GPU图片个数 | 学习率策略 |推理时间(fps) | Box AP | 下载 | 配置文件 | -| :------------------- | :------- | :-----: | :-----: | :------------: | :-----: | :-----------------------------------------------------: | :-----: | -| DarkNet53(paper) | 608 | 8 | 270e | ---- | 33.0 | - | - | -| DarkNet53(paper) | 416 | 8 | 270e | ---- | 31.0 | - | - | -| DarkNet53(paper) | 320 | 8 | 270e | ---- | 28.2 | - | - | -| DarkNet53 | 608 | 8 | 270e | ---- | 39.0 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_darknet53_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_darknet53_270e_coco.yml) | -| DarkNet53 | 416 | 8 | 270e | ---- | 37.5 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_darknet53_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_darknet53_270e_coco.yml) | -| DarkNet53 | 320 | 8 | 270e | ---- | 34.6 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_darknet53_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_darknet53_270e_coco.yml) | -| MobileNet-V1 | 608 | 8 | 270e | ---- | 28.8 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v1_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v1_270e_coco.yml) | -| MobileNet-V1 | 416 | 8 | 270e | ---- | 28.7 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v1_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v1_270e_coco.yml) | -| MobileNet-V1 | 320 | 8 | 270e | ---- | 26.5 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v1_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v1_270e_coco.yml) | -| MobileNet-V3 | 608 | 8 | 270e | ---- | 31.4 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v3_large_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v3_large_270e_coco.yml) | -| MobileNet-V3 | 416 | 8 | 270e | ---- | 29.7 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v3_large_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v3_large_270e_coco.yml) | -| MobileNet-V3 | 320 | 8 | 270e | ---- | 26.9 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v3_large_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v3_large_270e_coco.yml) | +### Cascade R-CNN -### YOLOv3 on Pasacl VOC +请参考[Cascade R-CNN](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/cascade_rcnn/) -| 骨架网络 | 输入尺寸 | 每张GPU图片个数 | 学习率策略 |推理时间(fps)| Box AP | 下载 | 配置文件 | -| :----------- | :--: | :-----: | :-----: |:------------: |:----: | :-------: | :----: | -| MobileNet-V1 | 608 | 8 | 270e | - | 75.1 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v1_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v1_270e_voc.yml) | -| MobileNet-V1 | 416 | 8 | 270e | - | 76.1 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v1_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v1_270e_voc.yml) | -| MobileNet-V1 | 320 | 8 | 270e | - | 73.6 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v1_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v1_270e_voc.yml) | -| MobileNet-V3 | 608 | 8 | 270e | - | 79.6 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v3_large_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v3_large_270e_voc.yml) | -| MobileNet-V3 | 416 | 8 | 270e | - | 78.6 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v3_large_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v3_large_270e_voc.yml) | -| MobileNet-V3 | 320 | 8 | 270e | - | 76.4 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/yolov3_mobilenet_v3_large_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3_mobilenet_v3_large_270e_voc.yml) | +### YOLOv3 -**注意:** YOLOv3均使用8GPU训练,训练270个epoch +请参考[YOLOv3](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/yolov3/) -### SSD on Pascal VOC +### SSD -| 骨架网络 | 网络类型 | 每张GPU图片个数 | 学习率策略 |推理时间(fps) | Box AP | 下载 | 配置文件 | -| :-------------- | :------------- | :-----: | :-----: | :------------: | :-----: | :-----------------------------------------------------: | :-----: | -| VGG | SSD | 8 | 240e | ---- | 78.2 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/ssd_vgg16_300_240e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/ssd_vgg16_300_240e_voc.yml) | - -**注意:** SSD使用4GPU训练,训练240个epoch +请参考[SSD](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/ssd/) ### FCOS -请参考[fcos](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/fcos/) +请参考[FCOS](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/fcos/) ### SOLOv2 -请参考[solov2](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/solov2/) +请参考[SOLOv2](https://github.com/PaddlePaddle/PaddleDetection/tree/dygraph/configs/solov2/) -- GitLab