提交 5431d5c4 编写于 作者: M minqiyang

Polish code

test=develop
上级 c35bf3d3
...@@ -149,7 +149,6 @@ def batched_multiclass_nms(boxes, scores, background, score_threshold, ...@@ -149,7 +149,6 @@ def batched_multiclass_nms(boxes, scores, background, score_threshold,
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 = boxes[n][idx][:] xmin, ymin, xmax, ymax = boxes[n][idx][:]
det_outs.append([c, scores[n][c][idx], xmin, ymin, xmax, ymax])
tmp_det_out.append( tmp_det_out.append(
[c, scores[n][c][idx], xmin, ymin, xmax, ymax]) [c, scores[n][c][idx], xmin, ymin, xmax, ymax])
sorted_det_out = sorted( sorted_det_out = sorted(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册