Merken Sie den Termin vor! Google I / O kehrt vom 18. bis 20. Mai zurück Registrieren Sie sich jetzt
Diese Seite wurde von der Cloud Translation API übersetzt.
Switch to English

Tensorflow :: ops :: NonMaxSuppressionV4

#include <image_ops.h>

Wählt gierig eine Teilmenge der Begrenzungsrahmen in absteigender Reihenfolge der Punktzahl aus.

Zusammenfassung

Das Wegschneiden von Feldern mit einer hohen Überschneidungsüberschneidung (IOU) überlappt sich mit zuvor ausgewählten Feldern. Begrenzungsrahmen mit einer Punktzahl von weniger als score_threshold werden entfernt. Begrenzungsrahmen werden als [y1, x1, y2, x2] geliefert, wobei (y1, x1) und (y2, x2) die Koordinaten eines beliebigen diagonalen Paares von Rahmenecken sind und die Koordinaten als normalisiert bereitgestellt werden können (dh in liegen) das Intervall [0, 1]) oder absolut. Es ist zu beachten, dass dieser Algorithmus unabhängig davon ist, wo sich der Ursprung im Koordinatensystem befindet, und allgemeiner für orthogonale Transformationen und Übersetzungen des Koordinatensystems unveränderlich ist. Das Verschieben oder Reflektieren des Koordinatensystems führt dazu, dass vom Algorithmus dieselben Kästchen ausgewählt werden. Die Ausgabe dieser Operation ist eine Reihe von Ganzzahlen, die in die Eingabesammlung von Begrenzungsrahmen indiziert werden, die die ausgewählten Felder darstellen. Die Begrenzungsrahmenkoordinaten, die den ausgewählten Indizes entsprechen, können dann unter Verwendung der tf.gather operation . Zum Beispiel: selected_indices = tf.image.non_max_suppression_v2 (Felder, Scores, max_output_size, iou_threshold, score_threshold) selected_boxes = tf.gather (Felder, selected_indices)

Argumente:

  • scope: Ein Scope- Objekt
  • Boxen: Ein 2-D-Float-Tensor mit Form [num_boxes, 4] .
  • Bewertungen: Ein 1-D-Float-Tensor der Form [num_boxes] der eine einzelne Bewertung darstellt, die jeder Box (jeder Reihe von Feldern) entspricht.
  • max_output_size: Ein skalarer ganzzahliger Tensor, der die maximale Anzahl von Feldern darstellt, die durch Nicht-Max-Unterdrückung ausgewählt werden sollen.
  • iou_threshold: Ein 0-D-Float-Tensor, der den Schwellenwert für die Entscheidung darstellt, ob sich Boxen in Bezug auf IOU zu stark überlappen.
  • score_threshold: Ein 0-D-Float-Tensor, der den Schwellenwert für die Entscheidung darstellt, wann Kästchen basierend auf der Punktzahl entfernt werden sollen.

Optionale Attribute (siehe Attrs ):

  • pad_to_max_output_size: Wenn true, wird die Ausgabe selected_indices so aufgefüllt, dass sie die Länge max_output_size . Der Standardwert ist false.

Kehrt zurück:

  • Output selected_indices: Ein 1-D-Integer-Tensor der Form [M] der die ausgewählten Indizes aus dem M <= max_output_size , wobei M <= max_output_size .
  • Output valid_outputs: Ein 0-D-Integer-Tensor, der die Anzahl der gültigen Elemente in selected_indices , wobei die gültigen Elemente zuerst angezeigt werden.

Konstruktoren und Destruktoren

NonMaxSuppressionV4 (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold)
NonMaxSuppressionV4 (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold, const NonMaxSuppressionV4::Attrs & attrs)

Öffentliche Attribute

operation
selected_indices
valid_outputs

Öffentliche statische Funktionen

PadToMaxOutputSize (bool x)

Strukturen

tensorflow :: ops :: NonMaxSuppressionV4 :: Attrs

Optionale Attributsetzer für NonMaxSuppressionV4 .

Öffentliche Attribute

Operation

Operation operation

selected_indices

::tensorflow::Output selected_indices

valid_outputs

::tensorflow::Output valid_outputs

Öffentliche Funktionen

NonMaxSuppressionV4

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

NonMaxSuppressionV4

 NonMaxSuppressionV4(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold,
  const NonMaxSuppressionV4::Attrs & attrs
)

Öffentliche statische Funktionen

PadToMaxOutputSize

Attrs PadToMaxOutputSize(
  bool x
)