tensorflow :: operaciones :: NonMaxSuppression

#include <image_ops.h>

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

Resumen

las cajas de poda que tienen una alta intersección sobre unión (IOU) se superponen con las cajas seleccionadas previamente. Los cuadros delimitadores se proporcionan como [y1, x1, y2, x2], donde (y1, x1) y (y2, x2) son las coordenadas de cualquier par diagonal de esquinas de cuadro y las coordenadas se pueden proporcionar como normalizadas (es decir, situadas en el intervalo [0, 1]) o absoluto. Tenga en cuenta que este algoritmo es independiente de dónde está el origen en el sistema de coordenadas. Tenga en cuenta que este algoritmo es invariante a las transformaciones ortogonales y traslaciones del sistema de coordenadas; por tanto, la traducción o las reflexiones del sistema de coordenadas dan como resultado que el algoritmo seleccione las mismas cajas. El resultado de esta operación es un conjunto de números enteros que se indexan 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: selected_indices = tf.image.non_max_suppression (cajas, puntuaciones, tamaño_de_salida_máx, iou_threshold) cajas_seleccionadas = tf.gather (cajas, índices_seleccionados)

Argumentos:

  • alcance: un objeto de alcance
  • cajas: Un tensor flotante 2-D de forma [num_boxes, 4] .
  • puntuaciones: Un tensor flotante 1-D de forma [num_boxes] representa una única puntuación correspondiente a cada casilla (cada fila de casillas).
  • max_output_size: un tensor entero escalar que representa el número máximo de casillas que se seleccionarán mediante la supresión no máxima.

Atributos opcionales (consulte Attrs ):

  • iou_threshold: un flotador que representa el umbral para decidir si los cuadros se superponen demasiado con respecto a los pagarés.

Devoluciones:

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

Constructores y Destructores

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)

Atributos públicos

operation
selected_indices

Funciones publicas

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

Funciones estáticas públicas

IouThreshold (float x)

Estructuras

tensorflow :: ops :: NonMaxSuppression :: Attrs

Establecedores de atributos opcionales para NonMaxSuppression .

Atributos públicos

operación

Operation operation

índices_seleccionados

::tensorflow::Output selected_indices

Funciones publicas

NonMaxSuppression

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

NonMaxSuppression

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

nodo

::tensorflow::Node * node() const 

operador :: tensorflow :: Entrada

 operator::tensorflow::Input() const 

operador :: tensorflow :: Salida

 operator::tensorflow::Output() const 

Funciones estáticas públicas

IouThreshold

Attrs IouThreshold(
  float x
)