Skip to content

Commit 7d2a54a

Browse files
nemonamelessnemonameless
authored
move ssld model (PaddlePaddle#2633)
* move ssld model, test=document_fix * fix typo, test=document_fix
1 parent bd5d850 commit 7d2a54a

19 files changed

+267
-57
lines changed

‎configs/cascade_rcnn/README.md

+4
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,10 @@
66
| :------------------- | :------------- | :-----: | :-----: | :------------: | :-----: | :-----: | :-----------------------------------------------------: | :-----: |
77
| ResNet50-FPN | Cascade Faster | 1 | 1x | ---- | 41.1 | - | [下载链接](https://paddledet.bj.bcebos.com/models/cascade_rcnn_r50_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco.yml) |
88
| ResNet50-FPN | Cascade Mask | 1 | 1x | ---- | 41.8 | 36.3 | [下载链接](https://paddledet.bj.bcebos.com/models/cascade_mask_rcnn_r50_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco.yml) |
9+
| ResNet50-vd-SSLDv2-FPN | Cascade Faster | 1 | 1x | ---- | 44.4 | - | [下载链接](https://paddledet.bj.bcebos.com/models/cascade_rcnn_r50_vd_fpn_ssld_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco.yml) |
10+
| ResNet50-vd-SSLDv2-FPN | Cascade Faster | 1 | 2x | ---- | 45.0 | - | [下载链接](https://paddledet.bj.bcebos.com/models/cascade_rcnn_r50_vd_fpn_ssld_2x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_2x_coco.yml) |
11+
| ResNet50-vd-SSLDv2-FPN | Cascade Mask | 1 | 1x | ---- | 44.9 | 39.1 | [下载链接](https://paddledet.bj.bcebos.com/models/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco.yml) |
12+
| ResNet50-vd-SSLDv2-FPN | Cascade Mask | 1 | 2x | ---- | 45.7 | 39.7 | [下载链接](https://paddledet.bj.bcebos.com/models/cascade_mask_rcnn_r50_vd_fpn_ssld_2x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_2x_coco.yml) |
913

1014
**注意：** Cascade R-CNN模型精度依赖Paddle develop分支修改，精度复现须使用[每日版本](https://www.paddlepaddle.org.cn/documentation/docs/zh/install/Tables.html#whl-dev)或2.0.1版本(将于2021.03发布)，使用Paddle 2.0.0版本会有少量精度损失。
1115

‎configs/ssld/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco.yml ‎configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco.yml

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
_BASE_: [
22
'../datasets/coco_instance.yml',
33
'../runtime.yml',
4-
'../cascade_rcnn/_base_/optimizer_1x.yml',
5-
'../cascade_rcnn/_base_/cascade_mask_rcnn_r50_fpn.yml',
6-
'../cascade_rcnn/_base_/cascade_mask_fpn_reader.yml',
4+
'_base_/optimizer_1x.yml',
5+
'_base_/cascade_mask_rcnn_r50_fpn.yml',
6+
'_base_/cascade_mask_fpn_reader.yml',
77
]
88
pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_ssld_v2_pretrained.pdparams
99
weights: output/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco/model_final

‎configs/ssld/cascade_mask_rcnn_r50_vd_fpn_ssld_2x_coco.yml ‎configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_2x_coco.yml

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
_BASE_: [
22
'../datasets/coco_instance.yml',
33
'../runtime.yml',
4-
'../cascade_rcnn/_base_/optimizer_1x.yml',
5-
'../cascade_rcnn/_base_/cascade_mask_rcnn_r50_fpn.yml',
6-
'../cascade_rcnn/_base_/cascade_mask_fpn_reader.yml',
4+
'_base_/optimizer_1x.yml',
5+
'_base_/cascade_mask_rcnn_r50_fpn.yml',
6+
'_base_/cascade_mask_fpn_reader.yml',
77
]
88
pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_ssld_v2_pretrained.pdparams
99
weights: output/cascade_mask_rcnn_r50_vd_fpn_ssld_2x_coco/model_final

‎configs/ssld/cascade_rcnn_r50_vd_fpn_ssld_1x_coco.yml ‎configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco.yml

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
_BASE_: [
22
'../datasets/coco_detection.yml',
33
'../runtime.yml',
4-
'../cascade_rcnn/_base_/optimizer_1x.yml',
5-
'../cascade_rcnn/_base_/cascade_rcnn_r50_fpn.yml',
6-
'../cascade_rcnn/_base_/cascade_fpn_reader.yml',
4+
'_base_/optimizer_1x.yml',
5+
'_base_/cascade_rcnn_r50_fpn.yml',
6+
'_base_/cascade_fpn_reader.yml',
77
]
88
pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_ssld_v2_pretrained.pdparams
99
weights: output/cascade_rcnn_r50_vd_fpn_ssld_1x_coco/model_final

‎configs/ssld/cascade_rcnn_r50_vd_fpn_ssld_2x_coco.yml ‎configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_2x_coco.yml

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
_BASE_: [
22
'../datasets/coco_detection.yml',
33
'../runtime.yml',
4-
'../cascade_rcnn/_base_/optimizer_1x.yml',
5-
'../cascade_rcnn/_base_/cascade_rcnn_r50_fpn.yml',
6-
'../cascade_rcnn/_base_/cascade_fpn_reader.yml',
4+
'_base_/optimizer_1x.yml',
5+
'_base_/cascade_rcnn_r50_fpn.yml',
6+
'_base_/cascade_fpn_reader.yml',
77
]
88
pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_ssld_v2_pretrained.pdparams
99
weights: output/cascade_rcnn_r50_vd_fpn_ssld_2x_coco/model_final

‎configs/faster_rcnn/README.md

+2
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,8 @@
1818
| ResNet101-vd-FPN | Faster | 1 | 2x | ---- | 43.0 | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_r101_vd_fpn_2x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_2x_coco.yml) |
1919
| ResNeXt101-vd-FPN | Faster | 1 | 1x | ---- | 43.4 | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_x101_vd_64x4d_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco.yml) |
2020
| ResNeXt101-vd-FPN | Faster | 1 | 2x | ---- | 44.0 | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_x101_vd_64x4d_fpn_2x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_2x_coco.yml) |
21+
| ResNet50-vd-SSLDv2-FPN | Faster | 1 | 1x | ---- | 41.4 | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_r50_vd_fpn_ssld_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_ssld_1x_coco.yml) |
22+
| ResNet50-vd-SSLDv2-FPN | Faster | 1 | 2x | ---- | 42.3 | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_r50_vd_ssld_fpn_2x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/faster_rcnn/faster_rcnn_r50_vd_ssld_fpn_2x_coco.yml) |
2123

2224
**注意：** Faster R-CNN模型精度依赖Paddle develop分支修改，精度复现须使用[每日版本](https://www.paddlepaddle.org.cn/documentation/docs/zh/install/Tables.html#whl-dev)或2.0.1版本(将于2021.03发布)，使用Paddle 2.0.0版本会有少量精度损失。
2325

‎configs/ssld/faster_rcnn_r50_vd_fpn_ssld_1x_coco.yml ‎configs/faster_rcnn/faster_rcnn_r50_vd_fpn_ssld_1x_coco.yml

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
_BASE_: [
22
'../datasets/coco_detection.yml',
33
'../runtime.yml',
4-
'../faster_rcnn/_base_/optimizer_1x.yml',
5-
'../faster_rcnn/_base_/faster_rcnn_r50_fpn.yml',
6-
'../faster_rcnn/_base_/faster_fpn_reader.yml',
4+
'_base_/optimizer_1x.yml',
5+
'_base_/faster_rcnn_r50_fpn.yml',
6+
'_base_/faster_fpn_reader.yml',
77
]
88
pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_ssld_v2_pretrained.pdparams
99
weights: output/faster_rcnn_r50_vd_fpn_ssld_1x_coco/model_final

‎configs/ssld/faster_rcnn_r50_vd_fpn_ssld_2x_coco.yml ‎configs/faster_rcnn/faster_rcnn_r50_vd_fpn_ssld_2x_coco.yml

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
_BASE_: [
22
'../datasets/coco_detection.yml',
33
'../runtime.yml',
4-
'../faster_rcnn/_base_/optimizer_1x.yml',
5-
'../faster_rcnn/_base_/faster_rcnn_r50_fpn.yml',
6-
'../faster_rcnn/_base_/faster_fpn_reader.yml',
4+
'_base_/optimizer_1x.yml',
5+
'_base_/faster_rcnn_r50_fpn.yml',
6+
'_base_/faster_fpn_reader.yml',
77
]
88
pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_ssld_v2_pretrained.pdparams
99
weights: output/faster_rcnn_r50_vd_fpn_ssld_2x_coco/model_final

‎configs/mask_rcnn/README.md

+2
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,8 @@
1414
| ResNet101-vd-FPN | Mask | 1 | 1x | ---- | 42.4 | 38.1 | [下载链接](https://paddledet.bj.bcebos.com/models/mask_rcnn_r101_vd_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco.yml) |
1515
| ResNeXt101-vd-FPN | Mask | 1 | 1x | ---- | 44.0 | 39.5 | [下载链接](https://paddledet.bj.bcebos.com/models/mask_rcnn_x101_vd_64x4d_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco.yml) |
1616
| ResNeXt101-vd-FPN | Mask | 1 | 2x | ---- | 44.6 | 39.8 | [下载链接](https://paddledet.bj.bcebos.com/models/mask_rcnn_x101_vd_64x4d_fpn_2x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_2x_coco.yml) |
17+
| ResNet50-vd-SSLDv2-FPN | Mask | 1 | 1x | ---- | 42.0 | 38.2 | [下载链接](https://paddledet.bj.bcebos.com/models/mask_rcnn_r50_vd_fpn_ssld_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_ssld_1x_coco.yml) |
18+
| ResNet50-vd-SSLDv2-FPN | Mask | 1 | 2x | ---- | 42.7 | 38.9 | [下载链接](https://paddledet.bj.bcebos.com/models/mask_rcnn_r50_vd_fpn_ssld_2x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_ssld_2x_coco.yml) |
1719

1820
**注意：** Mask R-CNN模型精度依赖Paddle develop分支修改，精度复现须使用[每日版本](https://www.paddlepaddle.org.cn/documentation/docs/zh/install/Tables.html#whl-dev)或2.0.1版本(将于2021.03发布)，使用Paddle 2.0.0版本会有少量精度损失。
1921

‎configs/ssld/mask_rcnn_r50_vd_fpn_ssld_1x_coco.yml ‎configs/mask_rcnn/mask_rcnn_r50_vd_fpn_ssld_1x_coco.yml

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
_BASE_: [
22
'../datasets/coco_instance.yml',
33
'../runtime.yml',
4-
'../mask_rcnn/_base_/optimizer_1x.yml',
5-
'../mask_rcnn/_base_/mask_rcnn_r50_fpn.yml',
6-
'../mask_rcnn/_base_/mask_fpn_reader.yml',
4+
'_base_/optimizer_1x.yml',
5+
'_base_/mask_rcnn_r50_fpn.yml',
6+
'_base_/mask_fpn_reader.yml',
77
]
88
pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_ssld_v2_pretrained.pdparams
99
weights: output/mask_rcnn_r50_vd_fpn_ssld_1x_coco/model_final

‎configs/ssld/mask_rcnn_r50_vd_fpn_ssld_2x_coco.yml ‎configs/mask_rcnn/mask_rcnn_r50_vd_fpn_ssld_2x_coco.yml

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
_BASE_: [
22
'../datasets/coco_instance.yml',
33
'../runtime.yml',
4-
'../mask_rcnn/_base_/optimizer_1x.yml',
5-
'../mask_rcnn/_base_/mask_rcnn_r50_fpn.yml',
6-
'../mask_rcnn/_base_/mask_fpn_reader.yml',
4+
'_base_/optimizer_1x.yml',
5+
'_base_/mask_rcnn_r50_fpn.yml',
6+
'_base_/mask_fpn_reader.yml',
77
]
88
pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_ssld_v2_pretrained.pdparams
99
weights: output/mask_rcnn_r50_vd_fpn_ssld_2x_coco/model_final

‎configs/ssld/README.md

-33
This file was deleted.

‎configs/yolov3/README.md

+9
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,9 @@
2222
| MobileNet-V3 | 608 | 8 | 270e | ---- | 31.4 | [下载链接](https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v3_large_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/yolov3/yolov3_mobilenet_v3_large_270e_coco.yml) |
2323
| MobileNet-V3 | 416 | 8 | 270e | ---- | 29.7 | [下载链接](https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v3_large_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/yolov3/yolov3_mobilenet_v3_large_270e_coco.yml) |
2424
| MobileNet-V3 | 320 | 8 | 270e | ---- | 26.9 | [下载链接](https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v3_large_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/yolov3/yolov3_mobilenet_v3_large_270e_coco.yml) |
25+
| MobileNet-V1-SSLD | 608 | 8 | 270e | ---- | 31.0 | [下载链接](https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v1_ssld_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/yolov3/yolov3_mobilenet_v1_ssld_270e_coco.yml) |
26+
| MobileNet-V1-SSLD | 416 | 8 | 270e | ---- | 30.6 | [下载链接](https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v1_ssld_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/yolov3/yolov3_mobilenet_v1_ssld_270e_coco.yml) |
27+
| MobileNet-V1-SSLD | 320 | 8 | 270e | ---- | 28.4 | [下载链接](https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v1_ssld_270e_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/yolov3/yolov3_mobilenet_v1_ssld_270e_coco.yml) |
2528

2629
### YOLOv3 on Pasacl VOC
2730

@@ -33,6 +36,12 @@
3336
| MobileNet-V3 | 608 | 8 | 270e | - | 79.6 | [下载链接](https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v3_large_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/yolov3/yolov3_mobilenet_v3_large_270e_voc.yml) |
3437
| MobileNet-V3 | 416 | 8 | 270e | - | 78.6 | [下载链接](https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v3_large_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/yolov3/yolov3_mobilenet_v3_large_270e_voc.yml) |
3538
| MobileNet-V3 | 320 | 8 | 270e | - | 76.4 | [下载链接](https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v3_large_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/yolov3/yolov3_mobilenet_v3_large_270e_voc.yml) |
39+
| MobileNet-V1-SSLD | 608 | 8 | 270e | - | 78.3 | [下载链接](https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v1_ssld_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/yolov3/yolov3_mobilenet_v1_ssld_270e_voc.yml) |
40+
| MobileNet-V1-SSLD | 416 | 8 | 270e | - | 79.6 | [下载链接](https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v1_ssld_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/yolov3/yolov3_mobilenet_v1_ssld_270e_voc.yml) |
41+
| MobileNet-V1-SSLD | 320 | 8 | 270e | - | 77.3 | [下载链接](https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v1_ssld_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/yolov3/yolov3_mobilenet_v1_ssld_270e_voc.yml) |
42+
| MobileNet-V3-SSLD | 608 | 8 | 270e | - | 80.4 | [下载链接](https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v3_large_ssld_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/yolov3/yolov3_mobilenet_v3_large_ssld_270e_voc.yml) |
43+
| MobileNet-V3-SSLD | 416 | 8 | 270e | - | 79.2 | [下载链接](https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v3_large_ssld_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/yolov3/yolov3_mobilenet_v3_large_ssld_270e_voc.yml) |
44+
| MobileNet-V3-SSLD | 320 | 8 | 270e | - | 77.3 | [下载链接](https://paddledet.bj.bcebos.com/models/yolov3_mobilenet_v3_large_ssld_270e_voc.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/develop/configs/yolov3/yolov3_mobilenet_v3_large_ssld_270e_voc.yml) |
3645

3746
**注意：** YOLOv3均使用8GPU训练，训练270个epoch
3847

‎configs/yolov3/yolov3_mobilenet_v1_ssld_270e_coco.yml

+11
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
_BASE_: [
2+
'../datasets/coco_detection.yml',
3+
'../runtime.yml',
4+
'_base_/optimizer_270e.yml',
5+
'_base_/yolov3_mobilenet_v1.yml',
6+
'_base_/yolov3_reader.yml',
7+
]
8+
9+
snapshot_epoch: 5
10+
pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/MobileNetV1_ssld_pretrained.pdparams
11+
weights: output/yolov3_mobilenet_v1_ssld_270e_coco/model_final

0 commit comments

Comments
 (0)