टेन्सरफ्लो::ऑप्स::नॉनमैक्ससप्रेशनV3
#include <image_ops.h>स्कोर के घटते क्रम में बाउंडिंग बॉक्स के एक उपसमूह का लालचपूर्वक चयन करता है।
सारांश
उन बॉक्सों को हटा दिया जाता है जिनका पहले से चुने गए बॉक्सों के साथ इंटरसेक्शन-ओवर-यूनियन (IOU) ओवरलैप अधिक होता है। score_threshold से कम स्कोर वाले बाउंडिंग बॉक्स हटा दिए जाते हैं। बाउंडिंग बॉक्स [y1, x1, y2, x2] के रूप में दिए जाते हैं, जहाँ (y1, x1) और (y2, x2) किसी भी विकर्ण कोने के जोड़े के निर्देशांक हैं और निर्देशांक सामान्यीकृत (अर्थात, अंतराल [0, 1] में स्थित) या निरपेक्ष रूप में दिए जा सकते हैं। ध्यान दें कि यह एल्गोरिदम निर्देशांक प्रणाली में मूल बिंदु की स्थिति से अप्रभावित है और सामान्यतः निर्देशांक प्रणाली के ऑर्थोगोनल रूपांतरणों और अनुवादों के प्रति अपरिवर्तनीय है; इस प्रकार निर्देशांक प्रणाली के अनुवाद या परावर्तन के परिणामस्वरूप एल्गोरिदम द्वारा समान बॉक्स चुने जाते हैं। इस ऑपरेशन का आउटपुट चयनित बॉक्सों का प्रतिनिधित्व करने वाले बाउंडिंग बॉक्सों के इनपुट संग्रह में अनुक्रमित पूर्णांकों का एक सेट है। चयनित सूचकांकों के अनुरूप बाउंडिंग बॉक्स निर्देशांक tf.gather operation उपयोग करके प्राप्त किए जा सकते हैं। उदाहरण के लिए: selected_indices = tf.image.non_max_suppression_v2( boxes, scores, max_output_size, iou_threshold, score_threshold) selected_boxes = tf.gather(boxes, selected_indices)
तर्क:
- स्कोप: एक स्कोप ऑब्जेक्ट
- बॉक्स:
[num_boxes, 4]आकार का एक 2-डी फ्लोट टेंसर। - स्कोर:
[num_boxes]आकार का एक 1-डी फ्लोट टेंसर जो प्रत्येक बॉक्स (बॉक्स की प्रत्येक पंक्ति) के अनुरूप एक एकल स्कोर का प्रतिनिधित्व करता है। - max_output_size: एक स्केलर पूर्णांक टेंसर जो नॉन मैक्स सप्रेशन द्वारा चयनित किए जाने वाले बॉक्सों की अधिकतम संख्या को दर्शाता है।
- iou_threshold: एक 0-D फ्लोट टेंसर जो IOU के संबंध में यह तय करने के लिए थ्रेशहोल्ड का प्रतिनिधित्व करता है कि क्या बॉक्स बहुत अधिक ओवरलैप करते हैं।
- score_threshold: एक 0-D फ्लोट टेंसर जो स्कोर के आधार पर बॉक्स को हटाने का निर्णय लेने के लिए निर्धारित सीमा का प्रतिनिधित्व करता है।
प्रतिफल:
-
Output:[M]आकार का एक 1-डी पूर्णांक टेंसर जो बॉक्स टेंसर से चयनित सूचकांकों का प्रतिनिधित्व करता है, जहाँ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
सार्वजनिक समारोह
नॉनमैक्ससप्रेशनवी3
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