diff --git a/dygraph/configs/dcn/README.md b/dygraph/configs/dcn/README.md index 8b272a25051ae0e9d5b46f9a249a9eb4d1880f99..6703af2314afb2d61acea7a211f8b0e6ec817fc2 100644 --- a/dygraph/configs/dcn/README.md +++ b/dygraph/configs/dcn/README.md @@ -2,14 +2,15 @@ | 骨架网络 | 网络类型 | 卷积 | 每张GPU图片个数 | 学习率策略 |推理时间(fps)| Box AP | Mask AP | 下载 | 配置文件 | | :------------------- | :------------- | :-----: |:--------: | :-----: | :-----------: |:----: | :-----: | :----------------------------------------------------------: | :----: | -| ResNet50-FPN | Faster | c3-c5 | 2 | 1x | - | 41.3 | - | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/faster_rcnn_dcn_r50_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/dcn/faster_rcnn_dcn_r50_fpn_1x_coco.yml) | -| ResNet50-vd-FPN | Faster | c3-c5 | 2 | 2x | - | 42.4 | - | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/faster_rcnn_dcn_r50_vd_fpn_2x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/dcn/faster_rcnn_dcn_r50_vd_fpn_2x_coco.yml) | -| ResNet101-vd-FPN | Faster | c3-c5 | 2 | 1x | - | 44.1 | - | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/faster_rcnn_dcn_r101_vd_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/dcn/faster_rcnn_dcn_r101_vd_fpn_1x_coco.yml) | -| ResNeXt101-vd-FPN | Faster | c3-c5 | 1 | 1x | - | 45.2 | - | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/faster_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.pdparams) |[配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/dcn/faster_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml) | -| ResNet50-FPN | Mask | c3-c5 | 1 | 1x | - | 41.9 | 37.3 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/mask_rcnn_dcn_r50_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/dcn/mask_rcnn_dcn_r50_fpn_1x_coco.yml) | -| ResNet50-vd-FPN | Mask | c3-c5 | 1 | 2x | - | 42.9 | 38.0 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/mask_rcnn_dcn_r50_vd_fpn_2x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/dcn/mask_rcnn_dcn_r50_vd_fpn_2x_coco.yml) | -| ResNet101-vd-FPN | Mask | c3-c5 | 1 | 1x | - | 44.6 | 39.2 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/mask_rcnn_dcn_r101_vd_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/dcn/mask_rcnn_dcn_r101_vd_fpn_1x_coco.yml) | -| ResNeXt101-vd-FPN | Mask | c3-c5 | 1 | 1x | - | 46.2 | 40.4 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/mask_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/dcn/mask_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml) | +| ResNet50-FPN | Faster | c3-c5 | 1 | 1x | - | 42.1 | - | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_dcn_r50_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/dcn/faster_rcnn_dcn_r50_fpn_1x_coco.yml) | +| ResNet50-vd-FPN | Faster | c3-c5 | 1 | 1x | - | 42.7 | - | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_dcn_r50_vd_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/dcn/faster_rcnn_dcn_r50_vd_fpn_1x_coco.yml) | +| ResNet50-vd-FPN | Faster | c3-c5 | 1 | 2x | - | 43.7 | - | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_dcn_r50_vd_fpn_2x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/dcn/faster_rcnn_dcn_r50_vd_fpn_2x_coco.yml) | +| ResNet101-vd-FPN | Faster | c3-c5 | 1 | 1x | - | 45.1 | - | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_dcn_r101_vd_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/dcn/faster_rcnn_dcn_r101_vd_fpn_1x_coco.yml) | +| ResNeXt101-vd-FPN | Faster | c3-c5 | 1 | 1x | - | 46.5 | - | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.pdparams) |[配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/dcn/faster_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml) | +| ResNet50-FPN | Mask | c3-c5 | 1 | 1x | - | - | - | [下载链接](https://paddledet.bj.bcebos.com/models/mask_rcnn_dcn_r50_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/dcn/mask_rcnn_dcn_r50_fpn_1x_coco.yml) | +| ResNet50-vd-FPN | Mask | c3-c5 | 1 | 2x | - | - | - | [下载链接](https://paddledet.bj.bcebos.com/models/mask_rcnn_dcn_r50_vd_fpn_2x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/dcn/mask_rcnn_dcn_r50_vd_fpn_2x_coco.yml) | +| ResNet101-vd-FPN | Mask | c3-c5 | 1 | 1x | - | - | - | [下载链接](https://paddledet.bj.bcebos.com/models/mask_rcnn_dcn_r101_vd_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/dcn/mask_rcnn_dcn_r101_vd_fpn_1x_coco.yml) | +| ResNeXt101-vd-FPN | Mask | c3-c5 | 1 | 1x | - | - | - | [下载链接](https://paddledet.bj.bcebos.com/models/mask_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/dcn/mask_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml) | **注意事项:** diff --git a/dygraph/configs/dcn/faster_rcnn_dcn_r101_vd_fpn_1x_coco.yml b/dygraph/configs/dcn/faster_rcnn_dcn_r101_vd_fpn_1x_coco.yml index 8ffb7e76d7a751a36012a57bf7d2d53c2e0719ec..1a69fc7f41c9eac37be38a508efede2e231c3d74 100644 --- a/dygraph/configs/dcn/faster_rcnn_dcn_r101_vd_fpn_1x_coco.yml +++ b/dygraph/configs/dcn/faster_rcnn_dcn_r101_vd_fpn_1x_coco.yml @@ -13,17 +13,3 @@ ResNet: return_idx: [0,1,2,3] num_stages: 4 dcn_v2_stages: [1,2,3] - -epoch: 12 -LearningRate: - base_lr: 0.02 - schedulers: - - !PiecewiseDecay - gamma: 0.1 - milestones: [8, 11] - - !LinearWarmup - start_factor: 0.1 - steps: 1000 - -TrainReader: - batch_size: 2 diff --git a/dygraph/configs/dcn/faster_rcnn_dcn_r50_fpn_1x_coco.yml b/dygraph/configs/dcn/faster_rcnn_dcn_r50_fpn_1x_coco.yml index 4540f51f719e800df7eddbe2c3be64dc6f3fbe41..1cd02ac1f2cedcc7892497752a4d4779dc635718 100644 --- a/dygraph/configs/dcn/faster_rcnn_dcn_r50_fpn_1x_coco.yml +++ b/dygraph/configs/dcn/faster_rcnn_dcn_r50_fpn_1x_coco.yml @@ -14,17 +14,3 @@ ResNet: return_idx: [0,1,2,3] num_stages: 4 dcn_v2_stages: [1,2,3] - -epoch: 12 -LearningRate: - base_lr: 0.02 - schedulers: - - !PiecewiseDecay - gamma: 0.1 - milestones: [8, 11] - - !LinearWarmup - start_factor: 0.1 - steps: 1000 - -TrainReader: - batch_size: 2 diff --git a/dygraph/configs/dcn/faster_rcnn_dcn_r50_vd_fpn_1x_coco.yml b/dygraph/configs/dcn/faster_rcnn_dcn_r50_vd_fpn_1x_coco.yml new file mode 100644 index 0000000000000000000000000000000000000000..4290f566b08721ca9a8daf08d76e34484afe4fa2 --- /dev/null +++ b/dygraph/configs/dcn/faster_rcnn_dcn_r50_vd_fpn_1x_coco.yml @@ -0,0 +1,15 @@ +_BASE_: [ + 'faster_rcnn_dcn_r50_fpn_1x_coco.yml', +] +pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/ResNet50_vd_pretrained.tar +weights: output/faster_rcnn_dcn_r50_vd_fpn_2x_coco/model_final + +ResNet: + # index 0 stands for res2 + depth: 50 + variant: d + norm_type: bn + freeze_at: 0 + return_idx: [0,1,2,3] + num_stages: 4 + dcn_v2_stages: [1,2,3] diff --git a/dygraph/configs/dcn/faster_rcnn_dcn_r50_vd_fpn_2x_coco.yml b/dygraph/configs/dcn/faster_rcnn_dcn_r50_vd_fpn_2x_coco.yml index 9269df4ad83a13817f4437f94360a37a0c744ce3..b49ce95bbced26a258582ea5eeba09556c7ea520 100644 --- a/dygraph/configs/dcn/faster_rcnn_dcn_r50_vd_fpn_2x_coco.yml +++ b/dygraph/configs/dcn/faster_rcnn_dcn_r50_vd_fpn_2x_coco.yml @@ -16,7 +16,7 @@ ResNet: epoch: 24 LearningRate: - base_lr: 0.02 + base_lr: 0.01 schedulers: - !PiecewiseDecay gamma: 0.1 @@ -24,6 +24,3 @@ LearningRate: - !LinearWarmup start_factor: 0.1 steps: 1000 - -TrainReader: - batch_size: 2 diff --git a/dygraph/configs/dcn/faster_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml b/dygraph/configs/dcn/faster_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml index d39c4cc216c3d503bdc2aeb46a26244c36ead8d0..e826d6afe02c46025a068928d7bf262d4770e9f6 100644 --- a/dygraph/configs/dcn/faster_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml +++ b/dygraph/configs/dcn/faster_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml @@ -16,14 +16,3 @@ ResNet: return_idx: [0,1,2,3] num_stages: 4 dcn_v2_stages: [1,2,3] - -epoch: 12 -LearningRate: - base_lr: 0.01 - schedulers: - - !PiecewiseDecay - gamma: 0.1 - milestones: [8, 11] - - !LinearWarmup - start_factor: 0.1 - steps: 1000 diff --git a/dygraph/configs/dcn/mask_rcnn_dcn_r101_vd_fpn_1x_coco.yml b/dygraph/configs/dcn/mask_rcnn_dcn_r101_vd_fpn_1x_coco.yml index 31738886842be13652fd0f0475b7ca0ee8ec74df..6413ccc53ddc017373c42f0d0276f2acae398ccb 100644 --- a/dygraph/configs/dcn/mask_rcnn_dcn_r101_vd_fpn_1x_coco.yml +++ b/dygraph/configs/dcn/mask_rcnn_dcn_r101_vd_fpn_1x_coco.yml @@ -13,14 +13,3 @@ ResNet: return_idx: [0,1,2,3] num_stages: 4 dcn_v2_stages: [1,2,3] - -epoch: 12 -LearningRate: - base_lr: 0.01 - schedulers: - - !PiecewiseDecay - gamma: 0.1 - milestones: [8, 11] - - !LinearWarmup - start_factor: 0.1 - steps: 1000 diff --git a/dygraph/configs/dcn/mask_rcnn_dcn_r50_fpn_1x_coco.yml b/dygraph/configs/dcn/mask_rcnn_dcn_r50_fpn_1x_coco.yml index d7db95dc0e2a64aa5eca21bd5e50f3326a1d60d6..b14a1ed1dd2ed3bf8522ebe810be8b0d4d0f80a7 100644 --- a/dygraph/configs/dcn/mask_rcnn_dcn_r50_fpn_1x_coco.yml +++ b/dygraph/configs/dcn/mask_rcnn_dcn_r50_fpn_1x_coco.yml @@ -14,14 +14,3 @@ ResNet: return_idx: [0,1,2,3] num_stages: 4 dcn_v2_stages: [1,2,3] - -epoch: 12 -LearningRate: - base_lr: 0.01 - schedulers: - - !PiecewiseDecay - gamma: 0.1 - milestones: [8, 11] - - !LinearWarmup - start_factor: 0.1 - steps: 1000 diff --git a/dygraph/configs/dcn/mask_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml b/dygraph/configs/dcn/mask_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml index ab8302e2e6901b3582c9644fd834f22be425ec68..72335c3c28ca1f772c900cfddcc77afbb2647b2d 100644 --- a/dygraph/configs/dcn/mask_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml +++ b/dygraph/configs/dcn/mask_rcnn_dcn_x101_vd_64x4d_fpn_1x_coco.yml @@ -16,14 +16,3 @@ ResNet: return_idx: [0,1,2,3] num_stages: 4 dcn_v2_stages: [1,2,3] - -epoch: 12 -LearningRate: - base_lr: 0.01 - schedulers: - - !PiecewiseDecay - gamma: 0.1 - milestones: [8, 11] - - !LinearWarmup - start_factor: 0.1 - steps: 1000 diff --git a/dygraph/configs/faster_rcnn/README.md b/dygraph/configs/faster_rcnn/README.md index b5c937cfa8afcf862019a2b7ab17374f834f4ed2..c7d261f100b55bb30fce08d76d3e2a7811c939ea 100644 --- a/dygraph/configs/faster_rcnn/README.md +++ b/dygraph/configs/faster_rcnn/README.md @@ -4,10 +4,20 @@ | 骨架网络 | 网络类型 | 每张GPU图片个数 | 学习率策略 |推理时间(fps) | Box 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/master/dygraph/configs/faster_rcnn/faster_rcnn_r50_1x_coco.yml) | -| ResNet34-vd-FPN | Faster | 2 | 1x | ---- | 37.5 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/faster_rcnn_r34_vd_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_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/master/dygraph/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.yml) | -| ResNeXt101-vd-FPN | Faster | 1 | 1x | ---- | 42.3 | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/faster_rcnn_x101_vd_64x4d_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco.yml) | +| ResNet50 | Faster | 1 | 1x | ---- | 36.7 | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_r50_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/faster_rcnn/faster_rcnn_r50_1x_coco.yml) | +| ResNet50-vd | Faster | 1 | 1x | ---- | 37.6 | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_r50_vd_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco.yml) | +| ResNet101 | Faster | 1 | 1x | ---- | 39.0 | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_r101_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/faster_rcnn/faster_rcnn_r101_1x_coco.yml) | +| ResNet34-FPN | Faster | 1 | 1x | ---- | 37.8 | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_r34_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco.yml) | +| ResNet34-vd-FPN | Faster | 1 | 1x | ---- | 38.5 | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_r34_vd_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco.yml) | +| ResNet50-FPN | Faster | 1 | 1x | ---- | 38.4 | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_r50_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.yml) | +| ResNet50-FPN | Faster | 1 | 2x | ---- | 40.0 | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_r50_fpn_2x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/faster_rcnn/faster_rcnn_r50_fpn_2x_coco.yml) | +| ResNet50-vd-FPN | Faster | 1 | 1x | ---- | 39.5 | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_r50_vd_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco.yml) | +| ResNet50-vd-FPN | Faster | 1 | 2x | ---- | 40.8 | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_r50_vd_fpn_2x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_2x_coco.yml) | +| ResNet101-FPN | Faster | 1 | 2x | ---- | 41.4 | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_r101_fpn_2x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/faster_rcnn/faster_rcnn_r101_fpn_2x_coco.yml) | +| ResNet101-vd-FPN | Faster | 1 | 1x | ---- | 42.0 | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_r101_vd_fpn_1x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco.yml) | +| 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/master/dygraph/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_2x_coco.yml) | +| 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/master/dygraph/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco.yml) | +| 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/master/dygraph/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_2x_coco.yml) | ## Citations ``` diff --git a/dygraph/configs/faster_rcnn/faster_rcnn_r101_fpn_2x_coco.yml b/dygraph/configs/faster_rcnn/faster_rcnn_r101_fpn_2x_coco.yml index 2ebbd04174e3110c7658ea56cb0b8016012798eb..540f1fe6a176f0d1508fe0e72d33bc3d634a6cac 100644 --- a/dygraph/configs/faster_rcnn/faster_rcnn_r101_fpn_2x_coco.yml +++ b/dygraph/configs/faster_rcnn/faster_rcnn_r101_fpn_2x_coco.yml @@ -21,5 +21,5 @@ LearningRate: gamma: 0.1 milestones: [16, 22] - !LinearWarmup - start_factor: 0.3333333333333333 - steps: 500 + start_factor: 0.1 + steps: 1000 diff --git a/dygraph/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco.yml b/dygraph/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco.yml index b40921f927eaa646ee32a814c8237f8f52330c53..38f564cef1fda078df6dea13f85ffa947667bbb7 100644 --- a/dygraph/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco.yml +++ b/dygraph/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco.yml @@ -12,14 +12,3 @@ ResNet: freeze_at: 0 return_idx: [0,1,2,3] num_stages: 4 - -epoch: 12 -LearningRate: - base_lr: 0.01 - schedulers: - - !PiecewiseDecay - gamma: 0.1 - milestones: [8, 11] - - !LinearWarmup - start_factor: 0.3333333333333333 - steps: 1000 diff --git a/dygraph/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_2x_coco.yml b/dygraph/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_2x_coco.yml index 5fc3953b056267c412316847c91eb5f467c960ab..8ac6aca88ea3580b30bed886a26bc498b733fa97 100644 --- a/dygraph/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_2x_coco.yml +++ b/dygraph/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_2x_coco.yml @@ -21,5 +21,5 @@ LearningRate: gamma: 0.1 milestones: [16, 22] - !LinearWarmup - start_factor: 0.3333333333333333 + start_factor: 0.1 steps: 1000 diff --git a/dygraph/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco.yml b/dygraph/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco.yml index 84d4d3b90db8c6e53a1d5be6990b92edad118554..b75bd40b31b9867f56310c3fcba05d9d31169748 100644 --- a/dygraph/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco.yml +++ b/dygraph/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco.yml @@ -12,23 +12,3 @@ ResNet: freeze_at: 0 return_idx: [0,1,2,3] num_stages: 4 - -FPN: - in_channels: [64, 128, 256, 512] - out_channel: 256 - min_level: 0 - max_level: 4 - spatial_scale: [0.25, 0.125, 0.0625, 0.03125] - -LearningRate: - base_lr: 0.02 - schedulers: - - !PiecewiseDecay - gamma: 0.1 - milestones: [8, 11] - - !LinearWarmup - start_factor: 0.1 - steps: 1000 - -TrainReader: - batch_size: 2 diff --git a/dygraph/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco.yml b/dygraph/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco.yml index bc24e1c92becbae41dc3a0e3450b4589bcf408d0..c8ae2a681e0476d3794addff5d32d69fcd964f2d 100644 --- a/dygraph/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco.yml +++ b/dygraph/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco.yml @@ -13,23 +13,3 @@ ResNet: freeze_at: 0 return_idx: [0,1,2,3] num_stages: 4 - -FPN: - in_channels: [64, 128, 256, 512] - out_channel: 256 - min_level: 0 - max_level: 4 - spatial_scale: [0.25, 0.125, 0.0625, 0.03125] - -LearningRate: - base_lr: 0.02 - schedulers: - - !PiecewiseDecay - gamma: 0.1 - milestones: [8, 11] - - !LinearWarmup - start_factor: 0.1 - steps: 1000 - -TrainReader: - batch_size: 2 diff --git a/dygraph/configs/faster_rcnn/faster_rcnn_r50_2x_coco.yml b/dygraph/configs/faster_rcnn/faster_rcnn_r50_2x_coco.yml deleted file mode 100644 index 0ce3a69b4abd1a7d85990ddeddce26ecd8548a72..0000000000000000000000000000000000000000 --- a/dygraph/configs/faster_rcnn/faster_rcnn_r50_2x_coco.yml +++ /dev/null @@ -1,15 +0,0 @@ -_BASE_: [ - 'faster_rcnn_r50_1x_coco.yml', -] -weights: output/faster_rcnn_r50_2x_coco/model_final - -epoch: 24 -LearningRate: - base_lr: 0.01 - schedulers: - - !PiecewiseDecay - gamma: 0.1 - milestones: [16, 22] - - !LinearWarmup - start_factor: 0.3333333333333333 - steps: 500 diff --git a/dygraph/configs/faster_rcnn/faster_rcnn_r50_fpn_2x_coco.yml b/dygraph/configs/faster_rcnn/faster_rcnn_r50_fpn_2x_coco.yml index f931515d3ab7ecc36741982e221d878d8988cc47..7edaadc30250ca5dec06f2db69650291027d4fd3 100644 --- a/dygraph/configs/faster_rcnn/faster_rcnn_r50_fpn_2x_coco.yml +++ b/dygraph/configs/faster_rcnn/faster_rcnn_r50_fpn_2x_coco.yml @@ -5,14 +5,11 @@ weights: output/faster_rcnn_r50_fpn_2x_coco/model_final epoch: 24 LearningRate: - base_lr: 0.02 + base_lr: 0.01 schedulers: - !PiecewiseDecay gamma: 0.1 milestones: [16, 22] - !LinearWarmup - start_factor: 0.3333333333333333 - steps: 500 - -TrainReader: - batch_size: 2 + start_factor: 0.1 + steps: 1000 diff --git a/dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco.yml b/dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco.yml new file mode 100644 index 0000000000000000000000000000000000000000..bbbf44c53c2e0c674296aa2975a8af439d1d6ece --- /dev/null +++ b/dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco.yml @@ -0,0 +1,14 @@ +_BASE_: [ + 'faster_rcnn_r50_fpn_1x_coco.yml', +] +pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/ResNet50_vd_pretrained.tar +weights: output/faster_rcnn_r50_vd_fpn_1x_coco/model_final + +ResNet: + # index 0 stands for res2 + depth: 50 + variant: d + norm_type: bn + freeze_at: 0 + return_idx: [0,1,2,3] + num_stages: 4 diff --git a/dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_2x_coco.yml b/dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_2x_coco.yml index 200ec47c23a1a4705953bbe92960f93dca807afc..34f496e3f9c6e5addf8171f5e28d790dcd770a85 100644 --- a/dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_2x_coco.yml +++ b/dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_2x_coco.yml @@ -15,14 +15,11 @@ ResNet: epoch: 24 LearningRate: - base_lr: 0.02 + base_lr: 0.01 schedulers: - !PiecewiseDecay gamma: 0.1 milestones: [16, 22] - !LinearWarmup - start_factor: 0.3333333333333333 - steps: 500 - -TrainReader: - batch_size: 2 + start_factor: 0.1 + steps: 1000 diff --git a/dygraph/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco.yml b/dygraph/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco.yml index 98631b9ea60a5be1ba64df29e393b699b4e693ab..68c2c4e8a422c5866c2db2264429031aa8fa233b 100644 --- a/dygraph/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco.yml +++ b/dygraph/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco.yml @@ -16,14 +16,3 @@ ResNet: freeze_at: 0 return_idx: [0,1,2,3] num_stages: 4 - -epoch: 12 -LearningRate: - base_lr: 0.01 - schedulers: - - !PiecewiseDecay - gamma: 0.1 - milestones: [8, 11] - - !LinearWarmup - start_factor: 0.1 - steps: 1000 diff --git a/dygraph/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_2x_coco.yml b/dygraph/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_2x_coco.yml new file mode 100644 index 0000000000000000000000000000000000000000..b4831719f06b5f547cc76e9cf7d5d9d03db4c158 --- /dev/null +++ b/dygraph/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_2x_coco.yml @@ -0,0 +1,29 @@ +_BASE_: [ + 'faster_rcnn_r50_fpn_1x_coco.yml', +] + +pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/ResNeXt101_vd_64x4d_pretrained.tar +weights: output/faster_rcnn_x101_vd_64x4d_fpn_2x_coco/model_final + +ResNet: + # for ResNeXt: groups, base_width, base_channels + depth: 101 + groups: 64 + base_width: 4 + base_channels: 64 + variant: d + norm_type: bn + freeze_at: 0 + return_idx: [0,1,2,3] + num_stages: 4 + +epoch: 24 +LearningRate: + base_lr: 0.01 + schedulers: + - !PiecewiseDecay + gamma: 0.1 + milestones: [16, 22] + - !LinearWarmup + start_factor: 0.1 + steps: 1000 diff --git a/dygraph/configs/gn/README.md b/dygraph/configs/gn/README.md index 5cf4cd4d1e790253d5a8782e50f33c07c0a6fcc0..510da624097f57d3164929ddefcce6d607476936 100644 --- a/dygraph/configs/gn/README.md +++ b/dygraph/configs/gn/README.md @@ -4,8 +4,8 @@ | 骨架网络 | 网络类型 | 每张GPU图片个数 | 学习率策略 |推理时间(fps)| Box AP | Mask AP | 下载 | 配置文件 | | :------------- | :------------- | :-----------: | :------: | :--------: |:-----: | :-----: | :----: | :----: | -| ResNet50-FPN | Faster | 1 | 2x | - | 41.9 | - | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/faster_rcnn_r50_fpn_gn_2x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/gn/faster_rcnn_r50_fpn_gn_2x_coco.yml) | -| ResNet50-FPN | Mask | 1 | 2x | - | - | - | [下载链接](https://paddlemodels.bj.bcebos.com/object_detection/dygraph/mask_rcnn_r50_fpn_gn_2x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/gn/mask_rcnn_r50_fpn_gn_2x_coco.yml) | +| ResNet50-FPN | Faster | 1 | 2x | - | 41.9 | - | [下载链接](https://paddledet.bj.bcebos.com/models/faster_rcnn_r50_fpn_gn_2x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/gn/faster_rcnn_r50_fpn_gn_2x_coco.yml) | +| ResNet50-FPN | Mask | 1 | 2x | - | 42.3 | 38.4 | [下载链接](https://paddledet.bj.bcebos.com/models/mask_rcnn_r50_fpn_gn_2x_coco.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/master/dygraph/configs/gn/mask_rcnn_r50_fpn_gn_2x_coco.yml) | **注意:** Faster R-CNN baseline仅使用 `2fc` head,而此处使用[`4conv1fc` head](https://arxiv.org/abs/1803.08494)(4层conv之间使用GN),并且FPN也使用GN,而对于Mask R-CNN是在mask head的4层conv之间也使用GN。