tensorflow:: عملیات:: NonMaxSuppressionV2

#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)

استدلال‌ها:

  • دامنه: یک شیء دامنه
  • جعبه‌ها: یک تانسور شناور دوبعدی به شکل [num_boxes, 4] .
  • نمرات: یک تانسور اعشاری یک بعدی به شکل [num_boxes] که نشان دهنده یک امتیاز واحد مربوط به هر جعبه (هر ردیف از جعبه‌ها) است.
  • max_output_size: یک تانسور عدد صحیح اسکالر که نشان‌دهنده حداکثر تعداد جعبه‌هایی است که باید با حذف غیر حداکثری انتخاب شوند.
  • iou_threshold: یک تانسور اعشاری صفر بعدی که نشان‌دهنده آستانه‌ای برای تصمیم‌گیری در مورد همپوشانی بیش از حد جعبه‌ها نسبت به IOU است.

بازده:

  • Output : یک تانسور عدد صحیح یک بعدی با شکل [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

شاخص‌های انتخاب‌شده

::tensorflow::Output selected_indices

کارکردهای عمومی

NonMaxSuppressionV2

 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