टेंसरफ़्लो :: ऑप्स :: NonMaxSuppression
#include <image_ops.h>
लालच स्कोर के अवरोही क्रम में बाउंडिंग बॉक्स के एक सबसेट का चयन करता है,।
सारांश
पहले से चयनित बक्से के साथ उच्च चौराहे-ओवर-यूनियन (IOU) ओवरलैप वाले बक्से को दूर करना। बाउंडिंग बॉक्स को [y1, X1, y2, x2] के रूप में आपूर्ति की जाती है, जहां (y1, X1) और (y2, x2) बॉक्स कोनों के किसी भी विकर्ण जोड़ी के निर्देशांक हैं और निर्देशांक सामान्यीकृत (यानी, झूठ बोलकर) प्रदान किए जा सकते हैं अंतराल [0, 1]) या निरपेक्ष। ध्यान दें कि यह एल्गोरिथ्म अज्ञेय है जहां मूल समन्वय प्रणाली में है। ध्यान दें कि यह एल्गोरिथ्म समन्वय प्रणाली के ऑर्थोगोनल परिवर्तनों और अनुवादों के लिए अपरिवर्तनीय है; इस प्रकार समन्वय प्रणाली के अनुवाद या प्रतिबिंब का परिणाम एल्गोरिदम द्वारा चुने गए समान बक्से में होता है। इस ऑपरेशन का आउटपुट चयनित बॉक्स का प्रतिनिधित्व करने वाले बाउंडिंग बॉक्स के इनपुट संग्रह में पूर्णांक अनुक्रमण का एक सेट है। बाउंडिंग बॉक्स चयनित सूचकांकों के अनुरूप है, फिर tf.gather operation
का उपयोग करके प्राप्त किया जा सकता है। उदाहरण के लिए: चयनित_इंडिस = tf.image.non_max_suppression (बॉक्स, स्कोर, अधिकतम_आउटपुट_साइज़, iou_threshold) चयनित_बॉक्स = tf.gather (बॉक्स, चयनित बिंदु)
तर्क:
- गुंजाइश: एक स्कोप ऑब्जेक्ट
- बक्से: आकार का एक 2-डी फ्लोट टेंसर
[num_boxes, 4]
। - स्कोर: आकार का एक 1-डी फ्लोट टेंसर
[num_boxes]
प्रत्येक बॉक्स (बॉक्स की प्रत्येक पंक्ति) के अनुरूप एकल स्कोर का प्रतिनिधित्व करता है। - max_output_size: एक स्केलर पूर्णांक टेन्सर, जो अधिकतम अधिकतम दमन द्वारा चुने जाने वाले बॉक्स की अधिकतम संख्या का प्रतिनिधित्व करता है।
वैकल्पिक विशेषताएँ ( Attrs
देखें):
- iou_threshold: यह तय करने के लिए थ्रेशोल्ड का प्रतिनिधित्व करने वाला एक फ्लोट है कि क्या बॉक्स IOU के संबंध में बहुत अधिक ओवरलैप करता है।
रिटर्न:
-
Output
: आकार का एक 1-डी पूर्णांक टेन्सर[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) |
संरचनाएं | |
---|---|
टेनसफ़्लो :: ऑप्स :: नॉनमैक्सअप्रेशन :: अट्र्स | 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
ऑपरेटर :: टेंसरफ़्लो :: इनपुट
operator::tensorflow::Input() constहै
ऑपरेटर :: टेंसरफ़्लो :: आउटपुट
operator::tensorflow::Output() const
सार्वजनिक स्थैतिक कार्य
IouThreshold
Attrs IouThreshold( float x )