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