tensorflow::ops::NonMaxSuppressionWithOverlaps

#include <image_ops.h>

เลือกกลุ่มย่อยของกรอบล้อมรอบโดยเรียงลำดับจากคะแนนมากที่สุดไปน้อยที่สุดอย่างโลภ (Greeding selects)

สรุป

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

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

selected_indices = tf.image.non_max_suppression_with_overlaps( overlaps, scores, max_output_size, overlap_threshold, score_threshold) selected_boxes = tf.gather(boxes, selected_indices)

ข้อโต้แย้ง:

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

การส่งคืนสินค้า:

  • 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

ดัชนีที่เลือก

::tensorflow::Output selected_indices

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

การระงับค่าสูงสุดที่ไม่ทับซ้อนกัน

 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 

ตัวดำเนินการ::เทนเซอร์โฟลว์::อินพุต

 operator::tensorflow::Input() const 

ตัวดำเนินการ::เทนเซอร์โฟลว์::เอาต์พุต

 operator::tensorflow::Output() const