nms

　

NMS

NMS 非极大值抑制

import tensorflow as tf
import numpy as np

rects=np.asarray([[1,2,3,4],[1,3,3,4], [1,3,4,4],[1,1,4,4],[1,1,3,4]],dtype=np.float32)
scores=np.asarray([0.4,0.5,0.72,0.9,0.45], dtype=np.float32)

with tf.session() as sess:
    nms = tf.image.non_max_suppression(rects, scores, max_output_size=5, iou_threshold=0.5)
    print(sess.run(nms))
    selected_boxes = sess.run(tf.gather(rects, nms))
    print(selected_boxes)

return

[3 2 0]
[[1. 1. 4. 4.]
 [1. 3. 4. 4.]
 [1. 2. 3. 4.]]

解释

# 返回极大值抑制后的indices
tf.image.non_max_suppression(）

# 根据返回的序列，得到最后返回的结果
tf.gather(rects, nms)

相关阅读

NMS(非极大值抑制)

NMS: non maximum suppression 翻译为“非极大值抑制”，为什么不翻译成最大值抑制呢？maximum可以翻译为“最大值”，也可以翻译成“极