fix nms unittest in py36, test=develop

revert-15774-anakin_subgraph_engine
jerrywgz 6 years ago
parent a0666b05c4
commit 4711d88a2f

@ -173,13 +173,16 @@ def lod_multiclass_nms(boxes, scores, background, score_threshold,
normalized,
shared=False)
if nmsed_num == 0:
#lod.append(1)
continue
lod.append(nmsed_num)
tmp_det_out = []
for c, indices in nmsed_outs.items():
for idx in indices:
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:
lod.append(1)

Loading…
Cancel
Save