tensorflow:: ops:: NonMaxSuppression

Selecciona de forma voraz un subconjunto de cuadros delimitadores en orden descendente de puntuación.

Resumen

Se eliminan las cajas que tienen una alta superposición de intersección sobre unión (IOU) con las cajas seleccionadas previamente. Las cajas delimitadoras se proporcionan como [y1, x1, y2, x2], donde (y1, x1) y (y2, x2) son las coordenadas de cualquier par diagonal de esquinas de la caja y las coordenadas se pueden proporcionar como normalizadas (es decir, que se encuentran en el intervalo [0, 1]) o absolutas. Tenga en cuenta que este algoritmo es independiente de dónde se encuentre el origen en el sistema de coordenadas. Tenga en cuenta que este algoritmo es invariante a las transformaciones ortogonales y las traslaciones del sistema de coordenadas; por lo tanto, las traslaciones o reflexiones del sistema de coordenadas dan como resultado que el algoritmo seleccione las mismas cajas. La salida de esta operación es un conjunto de enteros que indexan la colección de entrada de cajas delimitadoras que representan las cajas seleccionadas. Las coordenadas de la caja delimitadora correspondientes a los índices seleccionados se pueden obtener luego usando la tf.gather operation . Por ejemplo: selected_indices = tf.image.non_max_suppression( boxes, scores, max_output_size, iou_threshold) selected_boxes = tf.gather(boxes, selected_indices)

Argumentos:

• ámbito: Un objeto de ámbito
• cajas: Un tensor flotante 2D con forma [num_boxes, 4] .
• puntuaciones: Un tensor flotante unidimensional de forma [num_boxes] que representa una única puntuación correspondiente a cada cuadro (cada fila de cuadros).
• max_output_size: Un tensor entero escalar que representa el número máximo de cuadros que se seleccionarán mediante la supresión sin valor máximo.

Atributos opcionales (ver Attrs ):

• iou_threshold: Un valor flotante que representa el umbral para decidir si las cajas se superponen demasiado con respecto a IOU.

Devoluciones:

• Output : Un tensor entero unidimensional de forma [M] que representa los índices seleccionados del tensor de cajas, donde M <= max_output_size .

Atributos públicos

Operation operation

::tensorflow::Output selected_indices

 NonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size
)

 NonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  const NonMaxSuppression::Attrs & attrs
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs IouThreshold(
  float x
)