Tensorflow上的GPU上是否进行了非最大抑制？

Question

### folder ID ###

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]