tensorflow:: ops:: NonMaxSuppressionV2
#include <image_ops.h>Chciwie wybiera podzbiór pól ograniczających w kolejności malejącej według wyniku.
Streszczenie
Przycinanie pól o wysokim współczynniku przenikania się z poprzednio wybranymi polami (IOU). Pola ograniczające są podawane jako [y1, x1, y2, x2], gdzie (y1, x1) i (y2, x2) to współrzędne dowolnej pary przekątnych narożników pola, a współrzędne mogą być podane jako znormalizowane (tj. leżące w przedziale [0, 1]) lub bezwzględne. Należy pamiętać, że ten algorytm jest niezależny od położenia początku układu współrzędnych. Należy zauważyć, że ten algorytm jest niezmienny względem transformacji ortogonalnych i translacji układu współrzędnych; zatem translacje lub odbicia układu współrzędnych powodują wybranie przez algorytm tych samych pól.
Wynikiem tej operacji jest zbiór liczb całkowitych indeksowanych do wejściowego zbioru pól ograniczających reprezentujących wybrane pola. Współrzędne pól ograniczających odpowiadające wybranym indeksom można następnie uzyskać za pomocą tf.gather operation . Na przykład:
selected_indices = tf.image.non_max_suppression_v2( pola, wyniki, maksymalny_rozmiar_wyjściowy, próg_iou) selected_boxes = tf.gather( pola, selected_indices)
Argumenty:
- zakres: obiekt zakresu
- pola: dwuwymiarowy tensor typu float o kształcie
[num_boxes, 4]. - wyniki: jednowymiarowy tensor zmiennoprzecinkowy o kształcie
[num_boxes]reprezentujący pojedynczy wynik odpowiadający każdemu polu (każdemu wierszowi pól). - max_output_size: Skalarny tensor całkowity reprezentujący maksymalną liczbę pól, które mogą zostać wybrane przez pomijanie inne niż maksymalne.
- iou_threshold: Tensor zmiennoprzecinkowy 0-D reprezentujący próg decydujący o tym, czy pola nakładają się zbyt mocno w odniesieniu do IOU.
Zwroty:
-
Output: Jednowymiarowy tensor całkowity o kształcie[M]reprezentujący wybrane indeksy z tensora pól, gdzieM <= max_output_size.
Konstruktory i destruktory | |
|---|---|
NonMaxSuppressionV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input iou_threshold) |
Atrybuty publiczne | |
|---|---|
operation | |
selected_indices | |
Funkcje publiczne | |
|---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const | |
Atrybuty publiczne
działanie
Operation operation
wybrane_indeksy
::tensorflow::Output selected_indices
Funkcje publiczne
NonMaxSuppressionV2
NonMaxSuppressionV2( const ::tensorflow::Scope & scope, ::tensorflow::Input boxes, ::tensorflow::Input scores, ::tensorflow::Input max_output_size, ::tensorflow::Input iou_threshold )
węzeł
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const