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

Minor Fix for PaddleDetection (#3257)

* minor fix for PaddleDetection
上级 83987d01
......@@ -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
......
......@@ -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
......
......@@ -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:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册