Skip to content

Commit f74aa66

Browse files
authored
fix cascade_mask_rcnn export model (PaddlePaddle#4088)
1 parent 2c2eaf9 commit f74aa66

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

configs/cascade_rcnn/README.md

-1
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111
| 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) |
1212
| 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) |
1313

14-
**注意:** 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版本会有少量精度损失。
1514

1615
## Citations
1716
```

ppdet/engine/export_utils.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,8 @@ def _dump_infer_config(config, path, image_shape, model):
122122
format(infer_arch) +
123123
'Please set TRT_MIN_SUBGRAPH in ppdet/engine/export_utils.py')
124124
os._exit(0)
125-
if 'Mask' in infer_arch:
125+
if 'mask_head' in config[config['architecture']] and config[config[
126+
'architecture']]['mask_head']:
126127
infer_cfg['mask'] = True
127128
label_arch = 'detection_arch'
128129
if infer_arch in KEYPOINT_ARCH:

0 commit comments

Comments
 (0)