tensorflow::ops::NonMaxSuppressionV3

#include <image_ops.h>

Puanlarına göre azalan sırada, sınırlayıcı kutuların bir alt kümesini açgözlü bir şekilde seçer.

Özet

Daha önce seçilen kutularla yüksek kesişim-birleşim (IOU) örtüşmesine sahip kutular elenir. Skoru score_threshold düşük olan sınırlayıcı kutular kaldırılır. Sınırlayıcı kutular [y1, x1, y2, x2] olarak verilir; burada (y1, x1) ve (y2, x2), kutu köşelerinin herhangi bir diyagonal çiftinin koordinatlarıdır ve koordinatlar normalize edilmiş (yani [0, 1] aralığında yer alan) veya mutlak olarak verilebilir. Bu algoritmanın, koordinat sisteminde orijinin nerede olduğuna duyarsız olduğunu ve daha genel olarak koordinat sisteminin ortogonal dönüşümlerine ve ötelemelerine karşı değişmez olduğunu unutmayın; bu nedenle koordinat sisteminin ötelemesi veya yansıması, algoritma tarafından aynı kutuların seçilmesine neden olur. Bu işlemin çıktısı, seçilen kutuları temsil eden giriş sınırlayıcı kutu koleksiyonuna indekslenen bir dizi tamsayıdır. Seçilen indekslere karşılık gelen sınırlayıcı kutu koordinatları daha sonra tf.gather operation kullanılarak elde edilebilir. Örneğin: selected_indices = tf.image.non_max_suppression_v2( boxes, scores, max_output_size, iou_threshold, score_threshold) selected_boxes = tf.gather(boxes, selected_indices)

Argümanlar:

  • kapsam: Bir Kapsam nesnesi
  • kutular: [num_boxes, 4] şeklinde 2 boyutlu kayan noktalı bir tensör.
  • Puanlar: Her kutuya (her kutu satırına) karşılık gelen tek bir puanı temsil eden [num_boxes] şeklinde 1 boyutlu kayan noktalı bir tensör.
  • max_output_size: Maksimum bastırma olmaksızın seçilebilecek maksimum kutu sayısını temsil eden skalar bir tamsayı tensörü.
  • iou_threshold: Kutuların IOU'ya göre çok fazla örtüşüp örtüşmediğine karar vermek için kullanılan eşiği temsil eden 0 boyutlu kayan noktalı bir tensör.
  • score_threshold: Puana bağlı olarak kutuların ne zaman kaldırılacağına karar vermek için kullanılan eşiği temsil eden 0 boyutlu kayan noktalı bir tensör.

İadeler:

  • Output : Kutular tensöründen seçilen indeksleri temsil eden [M] şeklinde 1 boyutlu bir tamsayı tensörü; burada M <= max_output_size .

Yapıcılar ve Yıkıcılar

NonMaxSuppressionV3 (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold)

Genel özellikler

operation
selected_indices

Kamusal işlevler

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Genel özellikler

operasyon

Operation operation

seçilen_indeksler

::tensorflow::Output selected_indices

Kamusal işlevler

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
)

düğüm

::tensorflow::Node * node() const 

operatör::tensorflow::Giriş

 operator::tensorflow::Input() const 

operatör::tensorflow::Çıktı

 operator::tensorflow::Output() const