提交 1b42de24 编写于 作者: K Kaipeng Deng 提交者: GitHub

fix voc_eval difficult as None (#3259)

上级 f86b4012
...@@ -81,7 +81,8 @@ def bbox_eval(results, ...@@ -81,7 +81,8 @@ def bbox_eval(results,
for i in range(len(gt_boxes)): for i in range(len(gt_boxes)):
gt_box = gt_boxes[i] gt_box = gt_boxes[i]
gt_label = gt_labels[i] gt_label = gt_labels[i]
difficult = difficults[i] difficult = None if difficults is None \
else difficults[i]
bbox_num = bbox_lengths[i] bbox_num = bbox_lengths[i]
bbox = bboxes[bbox_idx: bbox_idx + bbox_num] bbox = bboxes[bbox_idx: bbox_idx + bbox_num]
gt_box, gt_label, difficult = prune_zero_padding( gt_box, gt_label, difficult = prune_zero_padding(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册