tensorflow:: ops:: NonMaxSuppressionWithOverlaps
#include <image_ops.h>Chciwie wybiera podzbiór pól ograniczających w kolejności malejącej według wyniku.
Streszczenie
Usuwanie pól o dużym nałożeniu na pola wcześniej zaznaczone. Pola ograniczające z wynikiem mniejszym niż score_threshold są usuwane. Wartości nakładania się Nxn są dostarczane w postaci macierzy kwadratowej, co pozwala na zdefiniowanie niestandardowego kryterium nakładania się (np. przecięcie przez sumę, przecięcie przez obszar itp.).
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_with_overlaps( nakładki, wyniki, maksymalny_rozmiar_wyjściowy, próg_nakładania, próg_wyników) selected_boxes = tf.gather(pola, selected_indices)
Argumenty:
- zakres: obiekt zakresu
- nakładanie się: 2-wymiarowy tensor zmiennoprzecinkowy o kształcie
[num_boxes, num_boxes]reprezentujący wartości nakładania się pól n na n. - 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.
- overlap_threshold: Tensor zmiennoprzecinkowy 0-D reprezentujący próg decydujący o tym, czy pola również na siebie nachodzą.
- score_threshold: Tensor zmiennoprzecinkowy 0-D reprezentujący próg decydujący o tym, kiedy usunąć pola na podstawie wyniku.
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 | |
|---|---|
NonMaxSuppressionWithOverlaps (const :: tensorflow::Scope & scope, :: tensorflow::Input overlaps, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input overlap_threshold, :: tensorflow::Input score_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
NonMaxSuppressionWithOverlaps
NonMaxSuppressionWithOverlaps( const ::tensorflow::Scope & scope, ::tensorflow::Input overlaps, ::tensorflow::Input scores, ::tensorflow::Input max_output_size, ::tensorflow::Input overlap_threshold, ::tensorflow::Input score_threshold )
węzeł
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const