tensorflow:: ops:: NonMaxSuppressionWithOverlaps

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

Resumen

Se eliminan los cuadros que se superponen en gran medida con los cuadros seleccionados previamente. Se eliminan los cuadros delimitadores con una puntuación inferior a score_threshold . Los valores de superposición de N x n se proporcionan como una matriz cuadrada, lo que permite definir un criterio de superposición personalizado (por ejemplo, intersección sobre unión, intersección sobre área, etc.).

El resultado de esta operación es un conjunto de números enteros que indexan la colección de cuadros delimitadores de entrada que representan los cuadros seleccionados. Las coordenadas de los cuadros delimitadores correspondientes a los índices seleccionados se pueden obtener mediante la tf.gather operation . Por ejemplo:

selected_indices = tf.image.non_max_suppression_with_overlaps( overlaps, scores, max_output_size, overlap_threshold, score_threshold) selected_boxes = tf.gather(boxes, selected_indices)

Argumentos:

• ámbito: Un objeto de ámbito
• superposiciones: Un tensor flotante 2D de forma [num_boxes, num_boxes] que representa los valores de superposición de cajas de n por n.
• 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.
• overlap_threshold: Un tensor flotante de 0 dimensiones que representa el umbral para decidir si las cajas también se superponen.
• score_threshold: Un tensor flotante de 0 dimensiones que representa el umbral para decidir cuándo eliminar cuadros en función de la puntuación.

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

 NonMaxSuppressionWithOverlaps(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input overlaps,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input overlap_threshold,
  ::tensorflow::Input score_threshold
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const