تينسورفلو :: العمليات :: 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 (مربعات ، مؤشرات_محددة)

الحجج:

  • النطاق: كائن النطاق
  • مربعات: موتر عائم ثنائي الأبعاد للشكل [num_boxes, 4] .
  • الدرجات: موتر عائم 1-D للشكل [num_boxes] يمثل درجة واحدة مقابلة لكل مربع (كل صف من المربعات).
  • max_output_size: موتر عدد صحيح عددي يمثل الحد الأقصى لعدد المربعات التي سيتم اختيارها عن طريق عدم الحد الأقصى للقمع.
  • iou_threshold: موتر عائم 0-D يمثل العتبة لتقرير ما إذا كانت الصناديق تتداخل كثيرًا فيما يتعلق بـ IOU.

عائدات:

  • Output : موتر عدد صحيح 1-D للشكل [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 :: Output

 operator::tensorflow::Output() const