टेंसरफ़्लो :: ऑप्स :: NonMaxSuppressionWithOverlaps
#include <image_ops.h>
लालच स्कोर के अवरोही क्रम में बाउंडिंग बॉक्स के एक सबसेट का चयन करता है,।
सारांश
पहले से चयनित बक्से के साथ उच्च ओवरलैप वाले बक्से को दूर करना। score_threshold
से कम स्कोर वाले बाउंडिंग बॉक्स हटा दिए score_threshold
हैं। एन-बाय-एन ओवरलैप मूल्यों को वर्ग मैट्रिक्स के रूप में आपूर्ति की जाती है, जो एक कस्टम ओवरलैप मानदंड (जैसे कि संघ पर चौराहा, क्षेत्र पर चौराहे, आदि) को परिभाषित करने की अनुमति देता है।
इस ऑपरेशन का आउटपुट चयनित बॉक्स का प्रतिनिधित्व करने वाले बाउंडिंग बॉक्स के इनपुट संग्रह में पूर्णांक अनुक्रमण का एक सेट है। बाउंडिंग बॉक्स चयनित सूचकांकों के अनुरूप है, फिर tf.gather operation
का उपयोग करके प्राप्त किया जा सकता है। उदाहरण के लिए:
चयनित_इंडिस = tf.image.non_max_suppression_with_overlaps (ओवरलैप्स, स्कोर, मैक्स_आउटपुट_साइज़, ओवरलैप_थ्रॉल्ड, स्कोर_थ्रॉल्ड) चयनित_बॉक्स = tf.gather (बक्से, चयनित_पिंड्स)
तर्क:
- गुंजाइश: एक स्कोप ऑब्जेक्ट
-
[num_boxes, num_boxes]
आकार का एक 2-डी फ्लोट टेंसर[num_boxes, num_boxes]
n-by-n बॉक्स ओवरलैप मूल्यों का प्रतिनिधित्व करता है। - स्कोर: आकृति का एक 1-डी फ्लोट टेंसर
[num_boxes]
प्रत्येक बॉक्स (बॉक्स की प्रत्येक पंक्ति) के अनुरूप एकल स्कोर का प्रतिनिधित्व करता है। - max_output_size: एक स्केलर पूर्णांक टेन्सर, जो अधिकतम अधिकतम दमन द्वारा चुने जाने वाले बॉक्स की अधिकतम संख्या को दर्शाता है।
- ओवरलैप_थ्रेशोल्ड: एक 0-डी फ्लोट टेंसर तय करने के लिए दहलीज का प्रतिनिधित्व करता है कि क्या बॉक्स भी ओवरलैप करते हैं।
- score_threshold: 0-D फ्लोट टेंसर स्कोर के आधार पर बक्से निकालने के लिए निर्णय लेने के लिए दहलीज का प्रतिनिधित्व करता है।
रिटर्न:
-
Output
: आकार का एक 1-डी पूर्णांक टेन्सर[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
ऑपरेटर :: टेंसरफ़्लो :: इनपुट
operator::tensorflow::Input() constहै
ऑपरेटर :: टेंसरफ़्लो :: आउटपुट
operator::tensorflow::Output() const