Google I/O एक लपेट है! TensorFlow सत्रों पर पकड़ बनाएं सत्र देखें

टेंसरफ़्लो :: ऑप्स :: NonMaxSuppressionV2

#include <image_ops.h>

लालच स्कोर के अवरोही क्रम में बाउंडिंग बॉक्स के एक सबसेट का चयन करता है,।

सारांश

पहले से चयनित बक्से के साथ उच्च चौराहे-ओवर-यूनियन (IOU) ओवरलैप वाले बक्से को दूर करना। बाउंडिंग बॉक्स को [y1, X1, y2, x2] के रूप में आपूर्ति की जाती है, जहां (y1, X1) और (y2, x2) बॉक्स कोनों के किसी भी विकर्ण जोड़ी के निर्देशांक हैं और निर्देशांक सामान्यीकृत (यानी, झूठ बोलकर) प्रदान किए जा सकते हैं अंतराल [0, 1]) या निरपेक्ष। ध्यान दें कि यह एल्गोरिथ्म अज्ञेय है जहां मूल समन्वय प्रणाली में है। ध्यान दें कि यह एल्गोरिथ्म समन्वय प्रणाली के ऑर्थोगोनल परिवर्तनों और अनुवादों के लिए अपरिवर्तनीय है; इस प्रकार समन्वय प्रणाली के अनुवाद या प्रतिबिंब का परिणाम एल्गोरिदम द्वारा चुने गए समान बक्से में होता है।

इस ऑपरेशन का आउटपुट चयनित बॉक्स का प्रतिनिधित्व करने वाले बाउंडिंग बॉक्स के इनपुट संग्रह में पूर्णांक अनुक्रमण का एक सेट है। बाउंडिंग बॉक्स चयनित सूचकांकों के अनुरूप है, फिर tf.gather operation का उपयोग करके प्राप्त किया जा सकता है। उदाहरण के लिए:

चुने हुए

तर्क:

  • गुंजाइश: एक स्कोप ऑब्जेक्ट
  • बक्से: आकार का एक 2-डी फ्लोट टेंसर [num_boxes, 4]
  • स्कोर: आकार का एक 1-डी फ्लोट टेंसर [num_boxes] प्रत्येक बॉक्स (बॉक्स की प्रत्येक पंक्ति) के अनुरूप एकल स्कोर का प्रतिनिधित्व करता है।
  • max_output_size: एक स्केलर पूर्णांक टेन्सर, जो अधिकतम अधिकतम दमन द्वारा चुने जाने वाले बॉक्स की अधिकतम संख्या का प्रतिनिधित्व करता है।
  • iou_threshold: 0-D फ्लोट टेंसर यह तय करने के लिए दहलीज का प्रतिनिधित्व करता है कि बॉक्स IOU के संबंध में बहुत अधिक ओवरलैप करते हैं या नहीं।

रिटर्न:

  • Output : आकार का एक 1-डी पूर्णांक टेन्सर [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 

ऑपरेटर :: टेंसरफ़्लो :: इनपुट

 operator::tensorflow::Input() const 
है

ऑपरेटर :: टेंसरफ़्लो :: आउटपुट

 operator::tensorflow::Output() const