未验证 提交 ae3e9a8b 编写于 作者: Y Yuan Gao 提交者: GitHub

Fix obj365 CACasacdeRCNN config and doc (#219)

* fix obj365 cascade rcnn config
* fix obj365 cascade rcnn doc
上级 d515d48d
...@@ -11,7 +11,7 @@ metric: COCO ...@@ -11,7 +11,7 @@ metric: COCO
num_classes: 501 num_classes: 501
CascadeRCNN: CascadeRCNN:
backbone: ResNet backbone: SENet
fpn: FPN fpn: FPN
rpn_head: FPNRPNHead rpn_head: FPNRPNHead
roi_extractor: FPNRoIAlign roi_extractor: FPNRoIAlign
...@@ -122,7 +122,7 @@ TrainReader: ...@@ -122,7 +122,7 @@ TrainReader:
image_dir: train image_dir: train
sample_transforms: sample_transforms:
- !DecodeImage - !DecodeImage
to_rgb: False to_rgb: True
- !RandomFlipImage - !RandomFlipImage
is_mask_flip: true is_mask_flip: true
is_normalized: false is_normalized: false
...@@ -197,7 +197,7 @@ EvalReader: ...@@ -197,7 +197,7 @@ EvalReader:
image_dir: val image_dir: val
sample_transforms: sample_transforms:
- !DecodeImage - !DecodeImage
to_rgb: False to_rgb: True
- !NormalizeImage - !NormalizeImage
is_channel_first: false is_channel_first: false
is_scale: False is_scale: False
...@@ -230,7 +230,7 @@ TestReader: ...@@ -230,7 +230,7 @@ TestReader:
anno_path: dataset/obj365/annotations/val.json anno_path: dataset/obj365/annotations/val.json
sample_transforms: sample_transforms:
- !DecodeImage - !DecodeImage
to_rgb: False to_rgb: True
- !NormalizeImage - !NormalizeImage
is_channel_first: false is_channel_first: false
is_scale: False is_scale: False
......
...@@ -38,7 +38,7 @@ python tools/train.py -c configs/obj365/cascade_rcnn_dcnv2_se154_vd_fpn_gn.yml ...@@ -38,7 +38,7 @@ python tools/train.py -c configs/obj365/cascade_rcnn_dcnv2_se154_vd_fpn_gn.yml
| 模型 | 验证集 mAP | 下载链接 | | 模型 | 验证集 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) |
## 模型效果 ## 模型效果
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册