未验证 提交 623f2ee4 编写于 作者: K Kaipeng Deng 提交者: GitHub

fix voc_eval difficult as None (#3259)

上级 f57d8a22
......@@ -81,7 +81,8 @@ def bbox_eval(results,
for i in range(len(gt_boxes)):
gt_box = gt_boxes[i]
gt_label = gt_labels[i]
difficult = difficults[i]
difficult = None if difficults is None \
else difficults[i]
bbox_num = bbox_lengths[i]
bbox = bboxes[bbox_idx: bbox_idx + bbox_num]
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.
先完成此消息的编辑!
想要评论请 注册