مؤتمر Google I / O هو التفاف! تابع جلسات TensorFlow اعرض الجلسات

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

#include <image_ops.h>

يختار جشعًا مجموعة فرعية من المربعات المحيطة بترتيب تنازلي للنتيجة.

ملخص

تشذيب المربعات التي تحتوي على تداخلات عالية مع المربعات المحددة مسبقًا. تتم إزالة score_threshold تحتوي على نقاط أقل من score_threshold . يتم توفير قيم التداخل N-by-n كمصفوفة مربعة ، مما يسمح بتعريف معيار التداخل المخصص (على سبيل المثال ، التقاطع على الاتحاد ، والتقاطع فوق المنطقة ، وما إلى ذلك).

ناتج هذه العملية عبارة عن مجموعة من الأعداد الصحيحة التي يتم فهرستها في مجموعة الإدخال من المربعات المحيطة التي تمثل المربعات المحددة. يمكن بعد ذلك الحصول على إحداثيات المربع المحيط المقابلة للمؤشرات المحددة باستخدام عملية tf.gather operation . على سبيل المثال:

selected_indices = tf.image.non_max_suppression_with_overlaps (التداخلات ، الدرجات ، max_output_size ، التراكب_threshold ، Score_threshold) selected_boxes = tf.gather (المربعات ، الفهارس المحددة)

الحجج:

  • النطاق: كائن النطاق
  • التداخلات: موتر عائم ثنائي الأبعاد للشكل [num_boxes, num_boxes] يمثل قيم متداخلة n-by-n box.
  • الدرجات: موتر عائم 1-D للشكل [num_boxes] يمثل درجة واحدة مناظرة لكل مربع (كل صف من المربعات).
  • max_output_size: موتر عدد صحيح عددي يمثل الحد الأقصى لعدد المربعات التي سيتم اختيارها عن طريق عدم الحد الأقصى للقمع.
  • interap_threshold: موتر عائم 0-D يمثل الحد الأدنى لتقرير ما إذا كانت المربعات تتداخل أيضًا.
  • Score_threshold: موتر عائم 0-D يمثل الحد الأدنى لتقرير وقت إزالة المربعات بناءً على الدرجة.

عائدات:

  • Output : موتر عدد صحيح 1-D للشكل [M] يمثل المؤشرات المحددة من موتر المربعات ، حيث M <= max_output_size .

البنائين والمدمرين

NonMaxSuppressionWithOverlaps (const :: tensorflow::Scope & scope, :: tensorflow::Input overlaps, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input overlap_threshold, :: tensorflow::Input score_threshold)

السمات العامة

operation
selected_indices

الوظائف العامة

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

السمات العامة

عملية

Operation operation

المؤشرات_المختارة

::tensorflow::Output selected_indices

الوظائف العامة

NonMaxSuppressionWithOverlaps

 NonMaxSuppressionWithOverlaps(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input overlaps,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input overlap_threshold,
  ::tensorflow::Input score_threshold
)

العقدة

::tensorflow::Node * node() const 

المشغل :: tensorflow :: الإدخال

 operator::tensorflow::Input() const 

المشغل :: Tensorflow :: Output

 operator::tensorflow::Output() const