tensorflow:: ops:: NonMaxSuppressionV2

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

Resumen

Se eliminan las cajas que presentan una alta superposición de intersección sobre unión (IOU) con las cajas previamente seleccionadas. Las cajas delimitadoras se proporcionan como [y1, x1, y2, x2], donde (y1, x1) y (y2, x2) son las coordenadas de cualquier par diagonal de vértices de la caja y pueden proporcionarse normalizadas (es decir, en el intervalo [0, 1]) o absolutas. Cabe destacar que este algoritmo es independiente de la ubicación del origen en el sistema de coordenadas. Asimismo, es invariante a las transformaciones ortogonales y a 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.

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_v2( 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 de 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.
• iou_threshold: Un tensor flotante de 0 dimensiones 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

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

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const