tensorflow:: ops:: NonMaxSuppressionV3
#include <image_ops.h>Chciwie wybiera podzbiór pól ograniczających w kolejności malejącej według wyniku.
Streszczenie
Przycinanie pól, które mają wysoki współczynnik przecięcia przez sumę (IOU) nakładający się na poprzednio wybrane pola. Pola ograniczające z wynikiem mniejszym niż score_threshold są usuwane. Pola ograniczające są dostarczane jako [y1, x1, y2, x2], gdzie (y1, x1) i (y2, x2) są współrzędnymi dowolnej pary przekątnej narożników pola, a współrzędne mogą być dostarczane jako znormalizowane (tj. leżące w przedziale [0, 1]) lub bezwzględne. Należy zauważyć, że ten algorytm jest agnostyczny względem położenia początku układu współrzędnych i, bardziej ogólnie, jest niezmienny względem przekształceń ortogonalnych i translacji układu współrzędnych; zatem translacje lub odbicia układu współrzędnych powodują, że te same pola są wybierane przez algorytm. Wynikiem tej operacji jest zbiór liczb całkowitych indeksowanych do zbioru wejściowego pól ograniczających reprezentujących wybrane pola. Współrzędne pola ograniczającego 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( boxes, scores, max_output_size, iou_threshold, score_threshold) selected_boxes = tf.gather(boxes, 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.
- 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 pudełek, gdzieM <= max_output_size.
Konstruktory i destruktory | |
|---|---|
NonMaxSuppressionV3 (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input iou_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
NonMaxSuppressionV3
NonMaxSuppressionV3( const ::tensorflow::Scope & scope, ::tensorflow::Input boxes, ::tensorflow::Input scores, ::tensorflow::Input max_output_size, ::tensorflow::Input iou_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