res/TensorFlowPythonExamples/examples/non_max_suppression_padded_2/__init__.py

   1 import tensorflow as tf
   2
   3 max_output_size = tf.compat.v1.constant(6)
   4 iou_threshold = tf.compat.v1.constant(0.5)
   5 score_threshold = tf.compat.v1.constant(0.6)
   6 pad_to_max_output_size = True
   7
   8 in_boxes_ = tf.compat.v1.placeholder(dtype=tf.float32, shape=(12, 4), name="Hole")
   9 in_scores_ = tf.compat.v1.placeholder(dtype=tf.float32, shape=(12), name="Hole")
  10
  11 non_max_suppression_padded_ = tf.compat.v1.image.non_max_suppression_padded(
  12     in_boxes_, in_scores_, max_output_size, iou_threshold, score_threshold,
  13     pad_to_max_output_size)