tensor akışı:: işlem:: NonMaxSuppressionV4
#include <image_ops.h>
Azalan puan sırasına göre sınırlayıcı kutuların bir alt kümesini açgözlülükle seçer.
Özet
daha önce seçilen kutularla çakışma oranı yüksek olan kutuların budanması. score_threshold
değerinden düşük puana sahip sınırlayıcı kutular kaldırılır. Sınırlayıcı kutular [y1, x1, y2, x2] olarak sağlanır; burada (y1, x1) ve (y2, x2), herhangi bir köşegen kutu köşesi çiftinin koordinatlarıdır ve koordinatlar normalleştirilmiş olarak sağlanabilir (yani, aralık [0, 1]) veya mutlak. Bu algoritmanın, orijinin koordinat sisteminde nerede olduğu konusunda agnostik olduğunu ve daha genel olarak koordinat sisteminin ortogonal dönüşümleri ve çevirileri için değişmez olduğunu unutmayın; dolayısıyla koordinat sisteminin çevrilmesi veya yansıtılması, algoritma tarafından aynı kutuların seçilmesine neden olur. Bu işlemin çıktısı, seçilen kutuları temsil eden sınırlayıcı kutuların giriş koleksiyonuna indekslenen bir tamsayılar kümesidir. Seçilen endekslere karşılık gelen sınırlayıcı kutu koordinatları daha sonra tf.gather operation
kullanılarak elde edilebilir. Örneğin: seçili_indices = tf.image.non_max_suppression_v2(boxs, puanlar, max_output_size, iou_threshold, Score_threshold) seçili_boxes = tf.gather(boxes, seçilmiş_indices)
Argümanlar:
- kapsam: Bir Kapsam nesnesi
- kutular:
[num_boxes, 4]
şeklinde bir 2 boyutlu kayan tensör. - puanlar: Her kutuya (kutuların her satırı) karşılık gelen tek bir puanı temsil eden
[num_boxes]
şeklindeki 1 boyutlu kayan tensör. - max_output_size: Maksimum dışı bastırma tarafından seçilecek maksimum kutu sayısını temsil eden bir skaler tamsayı tensörü.
- iou_threshold: Kutuların IOU'ya göre çok fazla örtüşüp örtüşmediğine karar vermek için eşiği temsil eden 0-D kayan tensör.
- Score_threshold: Puana göre kutuların ne zaman kaldırılacağına karar verme eşiğini temsil eden 0 boyutlu kayan tensör.
İsteğe bağlı özellikler (bkz. Attrs
):
- pad_to_max_output_size: Doğruysa,
selected_indices
çıktısımax_output_size
uzunluğunda olacak şekilde doldurulur. Varsayılan olarak false olur.
İade:
-
Output
seçili_indisler: Kutu tensöründen seçilen endeksleri temsil eden[M]
şeklinde bir 1 boyutlu tamsayı tensörü; buradaM <= max_output_size
. -
Output
valid_outputs:selected_indices
geçerli öğelerin sayısını temsil eden ve geçerli öğelerin ilk önce göründüğü 0 boyutlu bir tamsayı tensörü.
Yapıcılar ve Yıkıcılar | |
---|---|
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) |
Genel özellikler | |
---|---|
operation | |
selected_indices | |
valid_outputs |
Genel statik işlevler | |
---|---|
PadToMaxOutputSize (bool x) |
Yapılar | |
---|---|
tensorflow:: ops:: NonMaxSuppressionV4:: Attrs | NonMaxSuppressionV4 için isteğe bağlı öznitelik ayarlayıcılar. |
Genel özellikler
operasyon
Operation operation
seçilen_indisler
::tensorflow::Output selected_indices
valid_outputs
::tensorflow::Output valid_outputs
Kamu işlevleri
NonMaxSuppressionV4
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
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 )
Genel statik işlevler
PadToMaxOutputSize
Attrs PadToMaxOutputSize( bool x )