tensorflow:: ops:: NonMaxSuppressionV4

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.

Atrybuty opcjonalne (patrz Attrs ):

• pad_to_max_output_size: Jeśli wartość true, selected_indices wyjściowe są uzupełniane do długości max_output_size . Domyślnie false.

Zwroty:

• Output selected_indices: jednowymiarowy tensor całkowity o kształcie [M] reprezentujący wybrane indeksy z tensora pól, gdzie M <= max_output_size .
• Output valid_outputs: Tensor całkowity 0-D reprezentujący liczbę prawidłowych elementów w selected_indices , przy czym prawidłowe elementy pojawiają się jako pierwsze.

Atrybuty publiczne

Operation operation

::tensorflow::Output selected_indices

::tensorflow::Output valid_outputs

 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
)

Attrs PadToMaxOutputSize(
  bool x
)