2016-09-04 8 views
5

Tensorflow Android demo एक एंड्रॉइड ऐप बनाने के लिए एक सभ्य आधार प्रदान करता है जो टेंसरफ्लो ग्राफ का उपयोग करता है, लेकिन मैं उस ऐप के लिए इसे पुन: पेश करने के तरीके पर अटक गया हूं जो छवि वर्गीकरण नहीं करता है। जैसा कि है, यह एक .pb फ़ाइल से प्रारंभ ग्राफ में लोड होता है और इन्हें संदर्भों को चलाने के लिए उपयोग करता है (और कोड इस तरह मानता है), लेकिन मैं जो करना चाहता हूं वह मेरा स्वयं का ग्राफ लोड करता है (एक .pb फ़ाइल से), और ग्राफ़ के इनपुट/आउटपुट को संभालने के तरीके के बारे में एक कस्टम कार्यान्वयन करें।टेन्सफोर्लो एंड्रॉइड डेमो: एक कस्टम ग्राफ लोड करें?

प्रश्न में आलेख उदारता के गहरे शिक्षण पाठ्यक्रम के Assignment 6 से है, एक आरएनएन जो पाठ उत्पन्न करने के लिए एलएसटीएम का उपयोग करता है। (मैंने इसे पहले से ही एक .pb फ़ाइल में जमे हुए कर दिया है।) हालांकि, एंड्रॉइड डेमो का कोड इस धारणा पर आधारित है कि वे एक छवि वर्गीकरण से निपट रहे हैं। अब तक मुझे पता चला है कि मुझे tensorflow.initializeTensorflow (TensorFlowImageListener में बुलाया गया पैरामीटर) में पारित पैरामीटर के मानों को बदलने की आवश्यकता होगी, लेकिन कई पैरामीटर छवि इनपुट (जैसे IMAGE_SIZE) के गुणों का प्रतिनिधित्व करते हैं, जो ग्राफ मैं देख रहा हूं लोड करने के लिए नहीं है। क्या इसका मतलब है कि मुझे मूल कोड बदलना होगा? आम तौर पर, मैं इस पूरे मुद्दे से कैसे संपर्क कर सकता हूं?

+0

कोई समाधान मिला? –

उत्तर

0

TensorFlow Serving को टेन्सफ़ोर्लो मॉडल लोड करने और सेवा करने के सामान्य तरीके के लिए देखें।

0

अच्छी खबर: हाल ही में यह आपके एंड्रॉइड ऐप में प्री-प्रशिक्षित टेंसरफ्लो मॉडल को एम्बेड करना बहुत आसान हो गया है। यहाँ अपने ब्लॉग पोस्ट देखें:
https://medium.com/@daj/using-a-pre-trained-tensorflow-model-on-android-e747831a3d6 (भाग 1) https://medium.com/@daj/using-a-pre-trained-tensorflow-model-on-android-part-2-153ebdd4c465 (भाग 2)

मेरे ब्लॉग पोस्ट एक बहुत अधिक विस्तार में चला जाता है, लेकिन सारांश में, तुम सब करने की जरूरत है:

  1. अपने build.gradle में संकलन org.tensorflow:tensorflow-android:+ निर्भरता शामिल करें।
  2. अपने मॉडल के साथ इंटरफेस करने के लिए जावा TensorFlowInferenceInterface कक्षा का उपयोग करें (किसी मूल कोड को संशोधित करने की आवश्यकता नहीं है)।

टेंसरफ्लो एंड्रॉइड डेमो ऐप को इस नए दृष्टिकोण का उपयोग करने के लिए अपडेट किया गया है। TensorFlowImageClassifier.recognizeImage देखें जहां यह TensorFlowInferenceInterface का उपयोग करता है।

आपको अभी भी ग्राफ में इनपुट और आउटपुट नोड्स के इनपुट और इनपुट के आकार जैसे कुछ विन्यास निर्दिष्ट करने की आवश्यकता होगी, लेकिन आपको उस जानकारी को from using TensorBoard से बाहर करने में सक्षम होना चाहिए, या प्रशिक्षण का निरीक्षण करना चाहिए स्क्रिप्ट।

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