提交 f86b4012 编写于 作者: W wangguanzhong 提交者: GitHub

Minor Fix for PaddleDetection (#3257)

* minor fix for PaddleDetection
上级 83987d01
...@@ -7,7 +7,7 @@ snapshot_iter: 10000 ...@@ -7,7 +7,7 @@ snapshot_iter: 10000
use_gpu: true use_gpu: true
log_smooth_window: 20 log_smooth_window: 20
save_dir: output 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 weights: output/faster_rcnn_se154_vd_fpn_s1x/model_final
metric: COCO metric: COCO
num_classes: 81 num_classes: 81
......
...@@ -7,7 +7,7 @@ snapshot_iter: 10000 ...@@ -7,7 +7,7 @@ snapshot_iter: 10000
use_gpu: true use_gpu: true
log_smooth_window: 20 log_smooth_window: 20
save_dir: output 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/ weights: output/mask_rcnn_se154_vd_fpn_s1x/model_final/
metric: COCO metric: COCO
num_classes: 81 num_classes: 81
......
...@@ -75,9 +75,7 @@ class CascadeRCNN(object): ...@@ -75,9 +75,7 @@ class CascadeRCNN(object):
assert mode in ['train', 'test'], \ assert mode in ['train', 'test'], \
"only 'train' and 'test' mode is supported" "only 'train' and 'test' mode is supported"
if mode == 'train': if mode == 'train':
required_fields = [ required_fields = ['gt_label', 'gt_box', 'is_crowd', 'im_info']
'gt_label', 'gt_box', 'gt_mask', 'is_crowd', 'im_info'
]
else: else:
required_fields = ['im_shape', 'im_info'] required_fields = ['im_shape', 'im_info']
for var in required_fields: for var in required_fields:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册