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, gdzie M <= 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