res/TensorFlowPythonExamples/examples/non_max_suppression_with_scores/__init__.py

   1 import tensorflow as tf
   2
   3 max_output_size = tf.compat.v1.constant(4)
   4
   5 in_boxes_ = tf.compat.v1.placeholder(dtype=tf.float32, shape=(8, 4), name="Hole")
   6 in_scores_ = tf.compat.v1.placeholder(dtype=tf.float32, shape=(8), name="Hole")
   7
   8 # non_max_suppression_with_scores requires TF 1.15+
   9 non_max_suppression_with_scores_ = tf.compat.v1.image.non_max_suppression_with_scores(
  10     in_boxes_, in_scores_, max_output_size)