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

#include <image_ops.h>

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

สรุป

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

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

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

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

  • pad_per_class: หากเป็นเท็จช่องเอาต์พุต nmsed คะแนนและชั้นเรียนจะถูกเพิ่ม / ตัดเป็น max_total_size ถ้าเป็นจริงช่องเอาต์พุต nmsed คะแนนและชั้นเรียนจะมีความยาว max_size_per_class * num_classes เว้นแต่จะเกิน max_total_size ซึ่งในกรณีนี้จะถูกตัดเป็น max_total_size ค่าเริ่มต้นเป็นเท็จ
  • clip_boxes: ถ้าเป็นจริงให้สมมติว่าพิกัดของกล่องอยู่ระหว่าง [0, 1] และตัดกล่องเอาต์พุตหากอยู่เกิน [0, 1] ถ้าเป็นเท็จอย่าทำการตัดและส่งออกพิกัดกล่องตามที่เป็นอยู่

ผลตอบแทน:

  • Output nmsed_boxes: A [batch_size, max_detections, 4] float32 tensor ที่มีกล่องที่ไม่ถูกระงับสูงสุด
  • Output nmsed_scores: A [batch_size, max_detections] float32 tensor ที่มีคะแนนสำหรับกล่อง
  • Output nmsed_classes: A [batch_size, max_detections] float32 tensor ที่มีคลาสสำหรับกล่อง
  • Output valid_detections: A [batch_size] int32 tensor ที่ระบุจำนวนการตรวจจับที่ถูกต้องต่อรายการแบทช์ เฉพาะรายการ num_detections [i] อันดับต้น ๆ ใน nms_boxes [i], nms_scores [i] และ nms_class [i] เท่านั้นที่ถูกต้อง ส่วนที่เหลือของรายการเป็นศูนย์การพายเรือ

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

CombinedNonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size_per_class, :: tensorflow::Input max_total_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold)
CombinedNonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size_per_class, :: tensorflow::Input max_total_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold, const CombinedNonMaxSuppression::Attrs & attrs)

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

nmsed_boxes
nmsed_classes
nmsed_scores
operation
valid_detections

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

ClipBoxes (bool x)
PadPerClass (bool x)

โครงสร้าง

tensorflow :: ops :: CombinedNonMaxSuppression :: Attrs

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

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

nmsed_boxes

::tensorflow::Output nmsed_boxes

nmsed_classes

::tensorflow::Output nmsed_classes

nmsed_scores

::tensorflow::Output nmsed_scores

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

Operation operation

valid_detections

::tensorflow::Output valid_detections

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

CombinedNonMaxSuppression

 CombinedNonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size_per_class,
  ::tensorflow::Input max_total_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold
)

CombinedNonMaxSuppression

 CombinedNonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size_per_class,
  ::tensorflow::Input max_total_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold,
  const CombinedNonMaxSuppression::Attrs & attrs
)

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

ClipBoxes

Attrs ClipBoxes(
  bool x
)

PadPerClass

Attrs PadPerClass(
  bool x
)