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

#include <image_ops.h>

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

สรุป

การตัดแต่งกล่องที่มีจุดตัดสูงเกินสหภาพ (IOU) ทับซ้อนกับกล่องที่เลือกไว้ก่อนหน้านี้ กล่อง score_threshold มีคะแนนน้อยกว่า score_threshold จะถูกลบออก กล่องขอบเขตถูกจัดให้เป็น [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 score_threshold) selected_boxes = tf.gather (กล่อง selected_indices)

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

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

แอตทริบิวต์เสริม (ดู Attrs ):

  • pad_to_max_output_size: ถ้าเป็นจริงเอาต์พุตที่ selected_indices max_output_size จะมีความยาว max_output_size ค่าเริ่มต้นเป็นเท็จ

ผลตอบแทน:

  • Output เลือก M <= max_output_size ดัชนี: เทนเซอร์จำนวนเต็ม 1 มิติ [M] แทนดัชนีที่เลือกจากกล่องเทนเซอร์โดยที่ M <= max_output_size
  • Output valid_outputs: เทนเซอร์จำนวนเต็ม 0-D แทนจำนวนองค์ประกอบที่ถูกต้องใน selected_indices โดยองค์ประกอบที่ถูกต้องจะปรากฏก่อน

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

NonMaxSuppressionV4 (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold)
NonMaxSuppressionV4 (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold, const NonMaxSuppressionV4::Attrs & attrs)

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

operation
selected_indices
valid_outputs

ฟังก์ชันคงที่สาธารณะ

PadToMaxOutputSize (bool x)

โครงสร้าง

tensorflow :: ops :: NonMaxSuppressionV4 :: Attrs

ตัวตั้งค่าแอตทริบิวต์ทางเลือกสำหรับ NonMaxSuppressionV4

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

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

Operation operation

selected_indices

::tensorflow::Output selected_indices

valid_outputs

::tensorflow::Output valid_outputs

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

NonMaxSuppressionV4.0

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

NonMaxSuppressionV4.0

 NonMaxSuppressionV4(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold,
  const NonMaxSuppressionV4::Attrs & attrs
)

ฟังก์ชันคงที่สาธารณะ

PadToMaxOutputSize

Attrs PadToMaxOutputSize(
  bool x
)