2016-12-19 12 views
7

पैदा करता है:कैसे tf.train.batch CIFAR-10 TensorFlow ट्यूटोरियल मैं निम्न पंक्ति का सामना करना पड़ा में एक बैच

images, label_batch = tf.train.batch(
     [image, label], 
     batch_size=batch_size, 
     num_threads=num_preprocess_threads, 
     capacity=min_queue_examples + 3 * batch_size) 

समारोह tf.train.batch() एक इनपुट केवल एक छवि और एक लेबल के रूप में लग रहा है। फिर यह एकाधिक छवियों के साथ एक बैच कैसे बनाता है?

उत्तर

7

यह जोड़ी [image, label] इनपुट में लेता है, हाँ, यह एक ही जोड़ी है। tf.train.batch, हालांकि, आंतरिक रूप से एक कतार बनाता है। num_threads धागे जोड़े को कतार में capacity तक पहुंचाएंगे।

images, label_batch वास्तव में, डेक्यू संचालन हैं।

याद रखें कि आप एक कम्प्यूटेशनल ग्राफ तय कर रहे हैं, इसलिए जोड़ी [image, label] ग्राफ के दो नोड्स और विभिन्न असली अपने प्रशिक्षण सेट की image, label की जोड़ी, इन नोड्स के माध्यम से प्रवाह होगा प्रतिनिधित्व करते हैं। इस तरह, tf.train.batch छवियों और लेबलों के प्रवाह को पकड़ सकता है और कतार भर सकता है।

+2

मैं वर्तमान छवियों को कैसे मुद्रित कर सकता हूं जिन्हें प्रशिक्षण के लिए डिस्काउंट किया जा रहा है? मैं यहां दिए गए फॉर्म में slim.learning.train का उपयोग कर रहा हूं जैसा कि https://github.com/tensorflow/models/blob/master/slim/slim_walkthough.ipynb –

संबंधित मुद्दे