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 795108ff9cf07990a3bf5eb860e1fe0cc7aa67a3..c3dd761e6637f568c32d30af2300d87fe5f600ac 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 0d9d948850877a789f47934f0b2d1c08deabdf95..4430055712504bfe79252c1614792d05fb46b89b 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 6f4c416484c1979c61951d9b9599243bbcea09e8..3ab9369dcaa9935ad2fa1d67b7b7d8bd5d41904a 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: