تدفق التوتر:: العمليات:: 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 (boxes، Selected_indices)

الحجج:

  • النطاق: كائن النطاق
  • الصناديق: موتر عائم ثنائي الأبعاد ذو شكل [num_boxes, 4] .
  • الدرجات: موتر عائم أحادي الأبعاد ذو شكل [num_boxes] يمثل درجة واحدة تتوافق مع كل مربع (كل صف من المربعات).
  • max_output_size: موتر عددي صحيح يمثل الحد الأقصى لعدد المربعات التي سيتم تحديدها عن طريق القمع غير الأقصى.
  • iou_threshold: موتر عائم ذو 0-D يمثل العتبة لتحديد ما إذا كانت الصناديق تتداخل أكثر من اللازم فيما يتعلق بـ 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

Selected_indices

::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