移动paddle_detection

2024-09-24 17:02:56 +08:00
parent 90a6d5ec75
commit 3438cf6e0e
2025 changed files with 11 additions and 11 deletions
--- a/services/paddle_services/paddle_detection/configs/res2net/README.md
+++ b/services/paddle_services/paddle_detection/configs/res2net/README.md
@@ -0,0 +1,37 @@
+# Res2Net
+
+## Introduction
+
+- Res2Net: A New Multi-scale Backbone Architecture: [https://arxiv.org/abs/1904.01169](https://arxiv.org/abs/1904.01169)
+
+```
+@article{DBLP:journals/corr/abs-1904-01169,
+  author    = {Shanghua Gao and
+               Ming{-}Ming Cheng and
+               Kai Zhao and
+               Xinyu Zhang and
+               Ming{-}Hsuan Yang and
+               Philip H. S. Torr},
+  title     = {Res2Net: {A} New Multi-scale Backbone Architecture},
+  journal   = {CoRR},
+  volume    = {abs/1904.01169},
+  year      = {2019},
+  url       = {http://arxiv.org/abs/1904.01169},
+  archivePrefix = {arXiv},
+  eprint    = {1904.01169},
+  timestamp = {Thu, 25 Apr 2019 10:24:54 +0200},
+  biburl    = {https://dblp.org/rec/bib/journals/corr/abs-1904-01169},
+  bibsource = {dblp computer science bibliography, https://dblp.org}
+}
+```
+
+
+## Model Zoo
+
+| Backbone                | Type           | Image/gpu | Lr schd | Inf time (fps) | Box AP | Mask AP |                           Download                           | Configs |
+| :---------------------- | :------------- | :-------: | :-----: | :------------: | :----: | :-----: | :----------------------------------------------------------: | :-----: |
+| Res2Net50-FPN            | Faster         |     2     |   1x    |     -     |  40.6  |    -    | [model](https://paddledet.bj.bcebos.com/models/faster_rcnn_res2net50_vb_26w_4s_fpn_1x_coco.pdparams) |  [config](https://github.com/PaddlePaddle/PaddleDetection/blob/develop/configs/res2net/faster_rcnn_res2net50_vb_26w_4s_fpn_1x_coco.yml)  |
+| Res2Net50-FPN            | Mask         |     2     |   2x    |     -     |  42.4  |    38.1    | [model](https://paddledet.bj.bcebos.com/models/mask_rcnn_res2net50_vb_26w_4s_fpn_2x_coco.pdparams) | [config](https://github.com/PaddlePaddle/PaddleDetection/blob/develop/configs/res2net/mask_rcnn_res2net50_vb_26w_4s_fpn_2x_coco.yml) |
+| Res2Net50-vd-FPN            | Mask         |     2     |   2x    |     -     |  42.6  |    38.1    | [model](https://paddledet.bj.bcebos.com/models/mask_rcnn_res2net50_vd_26w_4s_fpn_2x_coco.pdparams) | [config](https://github.com/PaddlePaddle/PaddleDetection/blob/develop/configs/res2net/mask_rcnn_res2net50_vd_26w_4s_fpn_2x_coco.yml) |
+
+Note: all the above models are trained with 8 gpus.
--- a/services/paddle_services/paddle_detection/configs/res2net/faster_rcnn_res2net50_vb_26w_4s_fpn_1x_coco.yml
+++ b/services/paddle_services/paddle_detection/configs/res2net/faster_rcnn_res2net50_vb_26w_4s_fpn_1x_coco.yml
@@ -0,0 +1,33 @@
+_BASE_: [
+  '../datasets/coco_detection.yml',
+  '../runtime.yml',
+  '../faster_rcnn/_base_/optimizer_1x.yml',
+  '../faster_rcnn/_base_/faster_rcnn_r50_fpn.yml',
+  '../faster_rcnn/_base_/faster_fpn_reader.yml',
+]
+pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/Res2Net50_26w_4s_pretrained.pdparams
+weights: output/faster_rcnn_res2net50_vb_26w_4s_fpn_1x_coco/model_final
+
+FasterRCNN:
+  backbone: Res2Net
+  neck: FPN
+  rpn_head: RPNHead
+  bbox_head: BBoxHead
+  # post process
+  bbox_post_process: BBoxPostProcess
+
+
+Res2Net:
+  # index 0 stands for res2
+  depth: 50
+  width: 26
+  scales: 4
+  norm_type: bn
+  freeze_at: 0
+  return_idx: [0,1,2,3]
+  num_stages: 4
+  variant: b
+
+
+TrainReader:
+  batch_size: 2
--- a/services/paddle_services/paddle_detection/configs/res2net/mask_rcnn_res2net50_vb_26w_4s_fpn_2x_coco.yml
+++ b/services/paddle_services/paddle_detection/configs/res2net/mask_rcnn_res2net50_vb_26w_4s_fpn_2x_coco.yml
@@ -0,0 +1,47 @@
+_BASE_: [
+  '../datasets/coco_instance.yml',
+  '../runtime.yml',
+  '../mask_rcnn/_base_/optimizer_1x.yml',
+  '../mask_rcnn/_base_/mask_rcnn_r50_fpn.yml',
+  '../mask_rcnn/_base_/mask_fpn_reader.yml',
+]
+pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/Res2Net50_26w_4s_pretrained.pdparams
+weights: output/mask_rcnn_res2net50_vb_26w_4s_fpn_2x_coco/model_final
+
+MaskRCNN:
+  backbone: Res2Net
+  neck: FPN
+  rpn_head: RPNHead
+  bbox_head: BBoxHead
+  mask_head: MaskHead
+  # post process
+  bbox_post_process: BBoxPostProcess
+  mask_post_process: MaskPostProcess
+
+
+Res2Net:
+  # index 0 stands for res2
+  depth: 50
+  width: 26
+  scales: 4
+  norm_type: bn
+  freeze_at: 0
+  return_idx: [0,1,2,3]
+  num_stages: 4
+  variant: b
+
+
+epoch: 24
+LearningRate:
+  base_lr: 0.01
+  schedulers:
+  - !PiecewiseDecay
+    gamma: 0.1
+    milestones: [16, 22]
+  - !LinearWarmup
+    start_factor: 0.3333333333333333
+    steps: 500
+
+
+TrainReader:
+  batch_size: 2
--- a/services/paddle_services/paddle_detection/configs/res2net/mask_rcnn_res2net50_vd_26w_4s_fpn_2x_coco.yml
+++ b/services/paddle_services/paddle_detection/configs/res2net/mask_rcnn_res2net50_vd_26w_4s_fpn_2x_coco.yml
@@ -0,0 +1,47 @@
+_BASE_: [
+  '../datasets/coco_instance.yml',
+  '../runtime.yml',
+  '../mask_rcnn/_base_/optimizer_1x.yml',
+  '../mask_rcnn/_base_/mask_rcnn_r50_fpn.yml',
+  '../mask_rcnn/_base_/mask_fpn_reader.yml',
+]
+pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/Res2Net50_vd_26w_4s_pretrained.pdparams
+weights: output/mask_rcnn_res2net50_vd_26w_4s_fpn_2x_coco/model_final
+
+MaskRCNN:
+  backbone: Res2Net
+  neck: FPN
+  rpn_head: RPNHead
+  bbox_head: BBoxHead
+  mask_head: MaskHead
+  # post process
+  bbox_post_process: BBoxPostProcess
+  mask_post_process: MaskPostProcess
+
+
+Res2Net:
+  # index 0 stands for res2
+  depth: 50
+  width: 26
+  scales: 4
+  norm_type: bn
+  freeze_at: 0
+  return_idx: [0,1,2,3]
+  num_stages: 4
+  variant: d
+
+
+epoch: 24
+LearningRate:
+  base_lr: 0.01
+  schedulers:
+  - !PiecewiseDecay
+    gamma: 0.1
+    milestones: [16, 22]
+  - !LinearWarmup
+    start_factor: 0.3333333333333333
+    steps: 500
+
+
+TrainReader:
+  batch_size: 2