From 9105cc0234e2278ce433e277c4c4d46cc08fab89 Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Tue, 3 Sep 2019 12:08:18 +0800 Subject: [PATCH] Minor Fix for PaddleDetection (#3257) * minor fix for PaddleDetection --- .../PaddleDetection/configs/faster_rcnn_se154_vd_fpn_s1x.yml | 2 +- .../PaddleDetection/configs/mask_rcnn_se154_vd_fpn_s1x.yml | 2 +- .../ppdet/modeling/architectures/cascade_rcnn.py | 4 +--- 3 files changed, 3 insertions(+), 5 deletions(-) diff --git a/PaddleCV/PaddleDetection/configs/faster_rcnn_se154_vd_fpn_s1x.yml b/PaddleCV/PaddleDetection/configs/faster_rcnn_se154_vd_fpn_s1x.yml index 795108ff..c3dd761e 100644 --- a/PaddleCV/PaddleDetection/configs/faster_rcnn_se154_vd_fpn_s1x.yml +++ b/PaddleCV/PaddleDetection/configs/faster_rcnn_se154_vd_fpn_s1x.yml @@ -7,7 +7,7 @@ snapshot_iter: 10000 use_gpu: true log_smooth_window: 20 save_dir: output -pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/SE154_vd_pretrained.tar +pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/SENet154_vd_pretrained.tar weights: output/faster_rcnn_se154_vd_fpn_s1x/model_final metric: COCO num_classes: 81 diff --git a/PaddleCV/PaddleDetection/configs/mask_rcnn_se154_vd_fpn_s1x.yml b/PaddleCV/PaddleDetection/configs/mask_rcnn_se154_vd_fpn_s1x.yml index 0d9d9488..44300557 100644 --- a/PaddleCV/PaddleDetection/configs/mask_rcnn_se154_vd_fpn_s1x.yml +++ b/PaddleCV/PaddleDetection/configs/mask_rcnn_se154_vd_fpn_s1x.yml @@ -7,7 +7,7 @@ snapshot_iter: 10000 use_gpu: true log_smooth_window: 20 save_dir: output -pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/SE154_vd_pretrained.tar +pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/SENet154_vd_pretrained.tar weights: output/mask_rcnn_se154_vd_fpn_s1x/model_final/ metric: COCO num_classes: 81 diff --git a/PaddleCV/PaddleDetection/ppdet/modeling/architectures/cascade_rcnn.py b/PaddleCV/PaddleDetection/ppdet/modeling/architectures/cascade_rcnn.py index 6f4c4164..3ab9369d 100644 --- a/PaddleCV/PaddleDetection/ppdet/modeling/architectures/cascade_rcnn.py +++ b/PaddleCV/PaddleDetection/ppdet/modeling/architectures/cascade_rcnn.py @@ -75,9 +75,7 @@ class CascadeRCNN(object): assert mode in ['train', 'test'], \ "only 'train' and 'test' mode is supported" if mode == 'train': - required_fields = [ - 'gt_label', 'gt_box', 'gt_mask', 'is_crowd', 'im_info' - ] + required_fields = ['gt_label', 'gt_box', 'is_crowd', 'im_info'] else: required_fields = ['im_shape', 'im_info'] for var in required_fields: -- GitLab