python - Tensorflow 上的 GPU 是否进行了非极大值抑制？

Question

tf.image.non_max_suppression 

Tensorflow提供了这种非极大值抑制的方法，不知道tensorflow是否提供GPU支持？

Answer 1

答案是否。如果你尝试:

boxes = np.random.uniform(size=(100, 4))
scores = np.random.uniform(size=(100,))

bx = tf.constant(boxes, dtype=tf.float32)
sc = tf.constant(scores, dtype=tf.float32)

with tf.device("gpu:0"):
    indx = tf.image.non_max_suppression(bx, sc, 1000)

你会看到:

    NotFoundError: No registered 'NonMaxSuppressionV3' OpKernel for GPU devices compatible with node {{node NonMaxSuppressionV3}} = NonMaxSuppressionV3[T=DT_FLOAT](dummy_input, dummy_input, dummy_input, dummy_input, dummy_input)
    .  Registered:  device='CPU'; T in [DT_HALF]
  device='CPU'; T in [DT_FLOAT]
 [Op:NonMaxSuppressionV3]