提交 4711d88a 编写于 作者: J jerrywgz

fix nms unittest in py36, test=develop

上级 a0666b05
...@@ -173,13 +173,16 @@ def lod_multiclass_nms(boxes, scores, background, score_threshold, ...@@ -173,13 +173,16 @@ def lod_multiclass_nms(boxes, scores, background, score_threshold,
normalized, normalized,
shared=False) shared=False)
if nmsed_num == 0: if nmsed_num == 0:
#lod.append(1)
continue continue
lod.append(nmsed_num) lod.append(nmsed_num)
tmp_det_out = []
for c, indices in nmsed_outs.items(): for c, indices in nmsed_outs.items():
for idx in indices: for idx in indices:
xmin, ymin, xmax, ymax = box[idx, c, :] xmin, ymin, xmax, ymax = box[idx, c, :]
det_outs.append([c, score[idx][c], xmin, ymin, xmax, ymax]) tmp_det_out.append([c, score[idx][c], xmin, ymin, xmax, ymax])
sorted_det_out = sorted(
tmp_det_out, key=lambda tup: tup[0], reverse=False)
det_outs.extend(sorted_det_out)
if len(lod) == 0: if len(lod) == 0:
lod.append(1) lod.append(1)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册