टेंसरफ़्लो:: ऑप्स:: नॉनमैक्ससप्रेशनV2
#include <image_ops.h>
लालच से स्कोर के अवरोही क्रम में बाउंडिंग बॉक्स का एक सबसेट चुनता है ,।
सारांश
पहले से चयनित बक्सों के साथ उच्च इंटरसेक्शन-ओवर-यूनियन (आईओयू) ओवरलैप वाले बक्सों को छांटना। बाउंडिंग बॉक्स की आपूर्ति [Y1, X1, Y2, X2] के रूप में की जाती है, जहां (Y1, X1) और (Y2, x2) बॉक्स कॉर्नर के किसी भी विकर्ण जोड़ी के निर्देशांक हैं और निर्देशांक को सामान्यीकृत किया जा सकता है (यानी, झूठ बोल रहा है। अंतराल [0, 1]) या पूर्ण। ध्यान दें कि यह एल्गोरिथ्म अज्ञेय है जहां मूल समन्वय प्रणाली में है। ध्यान दें कि यह एल्गोरिथ्म ऑर्थोगोनल परिवर्तनों और समन्वय प्रणाली के अनुवादों के लिए अपरिवर्तनीय है; इस प्रकार समन्वय प्रणाली के अनुवाद या प्रतिबिंबों को एल्गोरिथ्म द्वारा चयनित समान बक्से में परिणाम दिया जाता है।
इस ऑपरेशन का आउटपुट चयनित बक्से का प्रतिनिधित्व करने वाले बाउंडिंग बॉक्स के इनपुट संग्रह में पूर्णांक अनुक्रमण का एक सेट है। बाउंडिंग बॉक्स चयनित सूचकांकों के अनुरूप निर्देशांक तब tf.gather operation
उपयोग करके प्राप्त किया जा सकता है। उदाहरण के लिए:
चयनित_इंडिस = tf.image.non_max_suppression_v2 (बॉक्स, स्कोर, MAX_OUTPUT_SIZE, IOU_THRESHOLD) CELECET_BOXES = TF.Gather (बक्से, चयनित_इंडिस)
तर्क:
- स्कोप: एक स्कोप ऑब्जेक्ट
- बक्से: एक 2-डी फ्लोट टेंसर आकार का
[num_boxes, 4]
। - स्कोर: 1-डी फ्लोट टेंसर ऑफ़ शेप
[num_boxes]
प्रत्येक बॉक्स (प्रत्येक पंक्ति की पंक्ति) के अनुरूप एकल स्कोर का प्रतिनिधित्व करता है। - MAX_OUTPUT_SIZE: गैर -मैक्स दमन द्वारा चुने जाने वाले बॉक्स की अधिकतम संख्या का प्रतिनिधित्व करने वाला एक स्केलर पूर्णांक टेंसर।
- IOU_THRESHOLD: एक 0-डी फ्लोट टेंसर यह तय करने के लिए दहलीज का प्रतिनिधित्व करता है कि क्या बक्से IOU के संबंध में बहुत अधिक ओवरलैप करते हैं।
रिटर्न:
-
Output
: आकार का एक 1-डी पूर्णांक टेंसर[M]
बॉक्स टेंसर से चयनित सूचकांकों का प्रतिनिधित्व करता है, जहांM <= max_output_size
।
निर्माता और विध्वंसक | |
---|---|
NonMaxSuppressionV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input iou_threshold) |
सार्वजनिक गुण | |
---|---|
operation | |
selected_indices |
सार्वजनिक समारोह | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
सार्वजनिक गुण
संचालन
Operation operation
चयनित_सूचकांक
::tensorflow::Output selected_indices
सार्वजनिक समारोह
नॉनमैक्ससप्रेशनV2
NonMaxSuppressionV2( const ::tensorflow::Scope & scope, ::tensorflow::Input boxes, ::tensorflow::Input scores, ::tensorflow::Input max_output_size, ::tensorflow::Input iou_threshold )
नोड
::tensorflow::Node * node() const
ऑपरेटर::टेन्सरफ़्लो::इनपुट
operator::tensorflow::Input() const
ऑपरेटर::टेन्सरफ़्लो::आउटपुट
operator::tensorflow::Output() const