تينسورفلو :: العمليات :: NonMaxSuppression

#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 (مربعات ، درجات ، max_output_size ، iou_threshold) selected_boxes = tf.gather (مربعات ، مؤشرات_مختارة)

الحجج:

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

السمات الاختيارية (انظر Attrs ):

  • iou_threshold: عائم يمثل الحد الأدنى لتقرير ما إذا كانت المربعات تتداخل كثيرًا مع IOU.

عائدات:

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

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

NonMaxSuppression

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

 operator::tensorflow::Output() const 

وظائف ثابتة عامة

IouThreshold

Attrs IouThreshold(
  float x
)