tensorflow:: ops:: דיכוי לא מקסימלי

#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(boxes, scores, max_output_size, iou_threshold) selected_boxes = tf.gather(boxes, selected_indices)

טיעונים:

  • היקף: אובייקט היקף
  • תיבות: טנזור צף דו-ממדי של צורה [num_boxes, 4] .
  • ניקוד: טנזור צף חד-ממדי בעל צורה [num_boxes] המייצג ניקוד יחיד המתאים לכל תיבה (כל שורה של תיבות).
  • max_output_size: טנזור סקלרי שלם המייצג את המספר המקסימלי של תיבות שייבחרו על ידי דיכוי שאינו מקסימלי.

מאפיינים אופציונליים (ראה Attrs ):

  • iou_threshold: ציפה המייצגת את הסף להחלטה האם תיבות חופפות יותר מדי ביחס ל-IOU.

החזרות:

  • Output : טנזור שלם חד-ממדי בעל צורה [M] המייצג את האינדקסים שנבחרו מטנזור התיבות, כאשר M <= max_output_size .

בנאים והורסים

NonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size)
NonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, const NonMaxSuppression::Attrs & attrs)

מאפיינים ציבוריים

operation
selected_indices

תפקידים ציבוריים

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

פונקציות סטטיות ציבוריות

IouThreshold (float x)

מבנים

tensorflow:: ops:: NonMaxSuppression:: Attrs

קובעי מאפיינים אופציונליים עבור NonMaxSuppression .

מאפיינים ציבוריים

מִבצָע 

Operation operation

מדדים_נבחרים 

::tensorflow::Output selected_indices

תפקידים ציבוריים

דיכוי לא מקסימלי 

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

דיכוי לא מקסימלי 

 NonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  const NonMaxSuppression::Attrs & attrs
)

צוֹמֶת 

::tensorflow::Node * node() const

אופרטור::tensorflow::קלט 

 operator::tensorflow::Input() const

אופרטור::tensorflow::פלט 

 operator::tensorflow::Output() const

פונקציות סטטיות ציבוריות

סף Iou 

Attrs IouThreshold(
  float x
)