From ae3e9a8b10bf8a7087615d7929f01ea317d23ee9 Mon Sep 17 00:00:00 2001 From: Yuan Gao Date: Mon, 10 Feb 2020 09:09:23 +0800 Subject: [PATCH] Fix obj365 CACasacdeRCNN config and doc (#219) * fix obj365 cascade rcnn config * fix obj365 cascade rcnn doc --- configs/obj365/cascade_rcnn_dcnv2_se154_vd_fpn_gn_cas.yml | 8 ++++---- docs/featured_model/CACascadeRCNN.md | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/configs/obj365/cascade_rcnn_dcnv2_se154_vd_fpn_gn_cas.yml b/configs/obj365/cascade_rcnn_dcnv2_se154_vd_fpn_gn_cas.yml index 4e9326e8d..b94797524 100644 --- a/configs/obj365/cascade_rcnn_dcnv2_se154_vd_fpn_gn_cas.yml +++ b/configs/obj365/cascade_rcnn_dcnv2_se154_vd_fpn_gn_cas.yml @@ -11,7 +11,7 @@ metric: COCO num_classes: 501 CascadeRCNN: - backbone: ResNet + backbone: SENet fpn: FPN rpn_head: FPNRPNHead roi_extractor: FPNRoIAlign @@ -122,7 +122,7 @@ TrainReader: image_dir: train sample_transforms: - !DecodeImage - to_rgb: False + to_rgb: True - !RandomFlipImage is_mask_flip: true is_normalized: false @@ -197,7 +197,7 @@ EvalReader: image_dir: val sample_transforms: - !DecodeImage - to_rgb: False + to_rgb: True - !NormalizeImage is_channel_first: false is_scale: False @@ -230,7 +230,7 @@ TestReader: anno_path: dataset/obj365/annotations/val.json sample_transforms: - !DecodeImage - to_rgb: False + to_rgb: True - !NormalizeImage is_channel_first: false is_scale: False diff --git a/docs/featured_model/CACascadeRCNN.md b/docs/featured_model/CACascadeRCNN.md index 30379c927..3a6d9720e 100644 --- a/docs/featured_model/CACascadeRCNN.md +++ b/docs/featured_model/CACascadeRCNN.md @@ -38,7 +38,7 @@ python tools/train.py -c configs/obj365/cascade_rcnn_dcnv2_se154_vd_fpn_gn.yml | 模型 | 验证集 mAP | 下载链接 | | :-----------------: | :--------: | :----------------------------------------------------------: | -| CACascadeRCNN SE154 | 31.7 | [model](https://paddlemodels.bj.bcebos.com/object_detection/cascade_rcnn_dcnv2_se154_vd_fpn_gn_cas_obj365.tar) | +| CACascadeRCNN SE154 | 31.6 | [model](https://paddlemodels.bj.bcebos.com/object_detection/cascade_rcnn_dcnv2_se154_vd_fpn_gn_cas_obj365.tar) | ## 模型效果 -- GitLab