flujo tensor:: operaciones:: Supresión no máxima con superposiciones

#include <image_ops.h>

Selecciona con avidez un subconjunto de cuadros delimitadores en orden descendente de puntuación.

Resumen

podando las cajas que tienen superposiciones altas con las cajas previamente seleccionadas. Se eliminan los cuadros delimitadores con una puntuación inferior a score_threshold . Los valores de superposición N por n se suministran 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 indexados en la colección de entrada de cuadros delimitadores que representan los cuadros seleccionados. Las coordenadas del cuadro delimitador correspondientes a los índices seleccionados se pueden obtener utilizando la tf.gather operation . Por ejemplo:

índices_seleccionados = tf.image.non_max_suppression_with_overlaps( superposiciones, puntuaciones, tamaño_salida_max, umbral_superposición, umbral_puntuación) cuadros_seleccionados = tf.gather(cuadros, índices_seleccionados)

Argumentos:

  • alcance: un objeto de alcance
  • superposiciones: un tensor flotante 2-D de forma [num_boxes, num_boxes] que representa los valores de superposición de cuadros n por n.
  • puntuaciones: un tensor flotante 1-D de forma [num_boxes] que representa una puntuación única 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 supresión no máxima.
  • overlay_threshold: un tensor flotante 0-D que representa el umbral para decidir si las cajas también se superponen.
  • score_threshold: tensor flotante 0-D que representa el umbral para decidir cuándo eliminar cuadros según la puntuación.

Devoluciones:

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

Constructores y destructores

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

Atributos públicos

operation
selected_indices

Funciones públicas

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Atributos públicos

operación

Operation operation

índices_seleccionados

::tensorflow::Output selected_indices

Funciones públicas

Supresión no máxima con superposiciones

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

nodo

::tensorflow::Node * node() const 

operador::tensorflow::Entrada

 operator::tensorflow::Input() const 

operador::tensorflow::Salida

 operator::tensorflow::Output() const