เทนเซอร์โฟลว์ :: ops :: NonMaxSuppressionV2.0

#include <image_ops.h>

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

สรุป

การตัดแต่งกล่องที่มีจุดตัดสูงเกินสหภาพ (IOU) ทับซ้อนกับกล่องที่เลือกไว้ก่อนหน้านี้ กล่องขอบเขตถูกจัดให้เป็น [y1, x1, y2, x2] โดยที่ (y1, x1) และ (y2, x2) เป็นพิกัดของมุมกล่องคู่แนวทแยง ช่วงเวลา [0, 1]) หรือค่าสัมบูรณ์ โปรดทราบว่าอัลกอริทึมนี้ไม่เชื่อเรื่องพระเจ้าว่าจุดเริ่มต้นอยู่ที่ใดในระบบพิกัด โปรดสังเกตว่าอัลกอริทึมนี้ไม่แปรผันกับการแปลงมุมฉากและการแปลของระบบพิกัด ดังนั้นการแปลหรือการสะท้อนของระบบพิกัดส่งผลให้อัลกอริทึมเลือกกล่องเดียวกัน

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

selected_indices = tf.image.non_max_suppression_v2 (กล่องคะแนน max_output_size iou_threshold) selected_boxes = tf.gather (กล่อง selected_indices)

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

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

ผลตอบแทน:

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

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

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

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

operation
selected_indices

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

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

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

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

Operation operation

selected_indices

::tensorflow::Output selected_indices

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

NonMaxSuppressionV2.0

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

โหนด

::tensorflow::Node * node() const 

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

 operator::tensorflow::Input() const 

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

 operator::tensorflow::Output() const