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