เทนเซอร์โฟลว์ :: ops :: NonMaxSuppressionWithOverlaps

#include <image_ops.h>

เลือกส่วนย่อยของกล่องขอบเขตตามลำดับคะแนนจากมากไปหาน้อย

สรุป

ตัดแต่งกล่องที่มีการทับซ้อนกันสูงกับกล่องที่เลือกไว้ก่อนหน้านี้ กล่อง score_threshold มีคะแนนน้อยกว่า score_threshold จะถูกลบออก ค่าการทับซ้อนแบบ N-by-n ถูกจัดให้เป็นเมทริกซ์สี่เหลี่ยมซึ่งช่วยในการกำหนดเกณฑ์การทับซ้อนที่กำหนดเอง (เช่นจุดตัดเหนือสหภาพการตัดกันบนพื้นที่ ฯลฯ )

ผลลัพธ์ของการดำเนินการนี้คือชุดของจำนวนเต็มที่ทำดัชนีในคอลเลกชันอินพุตของกล่องขอบเขตที่แสดงถึงกล่องที่เลือก จากนั้นสามารถหาพิกัดกล่องขอบเขตที่สอดคล้องกับดัชนีที่เลือกได้โดยใช้การ tf.gather operation ตัวอย่างเช่น:

selected_indices = tf.image.non_max_suppression_with_overlaps (คาบเกี่ยวกันคะแนน max_output_size เหลื่อม _threshold score_threshold) selected_boxes = tf.gather (กล่อง selected_indices)

อาร์กิวเมนต์:

  • ขอบเขต: วัตถุ ขอบเขต
  • การทับซ้อนกัน: เทนเซอร์ลอยตัว 2 มิติของรูปร่าง [num_boxes, num_boxes] แทนค่าการทับซ้อนของกล่อง n-by-n
  • คะแนน: รูปร่างเทนเซอร์ลอยตัว 1 มิติ [num_boxes] แสดงถึงคะแนนเดียวที่สอดคล้องกับแต่ละกล่อง (แต่ละแถวของกล่อง)
  • max_output_size: เทนเซอร์จำนวนเต็มสเกลาร์ที่แสดงถึงจำนวนกล่องสูงสุดที่จะเลือกโดยไม่ใช่การป้องกันสูงสุด
  • overlap_threshold: เทนเซอร์ลอย 0-D แสดงถึงขีด จำกัด สำหรับการตัดสินใจว่ากล่องทับซ้อนกันเกินไปหรือไม่
  • score_threshold: เทนเซอร์ลอยตัว 0-D แสดงถึงเกณฑ์สำหรับการตัดสินใจว่าจะลบกล่องตามคะแนนเมื่อใด

ผลตอบแทน:

  • Output : เทนเซอร์จำนวนเต็ม 1 มิติ [M] แทนดัชนีที่เลือกจากกล่องเทนเซอร์โดยที่ M <= max_output_size

ผู้สร้างและผู้ทำลาย

NonMaxSuppressionWithOverlaps (const :: tensorflow::Scope & scope, :: tensorflow::Input overlaps, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input overlap_threshold, :: tensorflow::Input score_threshold)

คุณลักษณะสาธารณะ

operation
selected_indices

หน้าที่สาธารณะ

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

คุณลักษณะสาธารณะ

การดำเนินการ

Operation operation

selected_indices

::tensorflow::Output selected_indices

หน้าที่สาธารณะ

NonMaxSuppressionWithOverlaps

 NonMaxSuppressionWithOverlaps(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input overlaps,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input overlap_threshold,
  ::tensorflow::Input score_threshold
)

โหนด

::tensorflow::Node * node() const 

ตัวดำเนินการ :: tensorflow :: อินพุต

 operator::tensorflow::Input() const 

ตัวดำเนินการ :: tensorflow :: เอาท์พุท

 operator::tensorflow::Output() const