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

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 .

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