tensorflow::ops::NonMaxSuppression

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

Önceden seçilen kutularla yüksek kesişim-birleşim (IOU) örtüşmesine sahip kutuları budama işlemi gerçekleştirilir. Sınır kutuları [y1, x1, y2, x2] olarak verilir; burada (y1, x1) ve (y2, x2), herhangi bir köşegen çiftinin koordinatlarıdır ve koordinatlar normalize edilmiş (yani [0, 1] aralığında yer alan) veya mutlak olarak verilebilir. Bu algoritmanın koordinat sistemindeki orijinin nerede olduğuna duyarsız olduğunu unutmayın. Bu algoritmanın, 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 sınır kutularının giriş koleksiyonuna indekslenen bir dizi tamsayıdır. Seçilen indekslere karşılık gelen sınır kutusu koordinatları daha sonra tf.gather operation kullanılarak elde edilebilir. Örneğin: selected_indices = tf.image.non_max_suppression( boxes, scores, max_output_size, iou_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ü.

İsteğe bağlı özellikler (bkz. Attrs ):

• 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 bir ondalık sayı.

İadeler:

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

Genel özellikler

Operation operation

::tensorflow::Output selected_indices

 NonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size
)

 NonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  const NonMaxSuppression::Attrs & attrs
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs IouThreshold(
  float x
)