2016-04-06 12 views
2

yüklenemedi. TensorFlow tutorial kullanarak kendi özel resimlerim için bir protobuf dosyası (.pb dosya) oluşturdum.Tensor Flow Özel protobuf dosyaları

Ama aynı dosyayı tensorflow/examples/android/assets öğesindeki varlık klasörüne değiştirdiğimde ve bir APK oluşturup oluşturmaya çalıştığınızda, APK oluşturulur, ancak APK'yı bir Android cihazında çalıştırdığımda, APK çöküyor.

Python'dan classify_image'u çalıştırırsam, bana uygun sonuçlar verir.

Herhangi bir yardımı için teşekkür ederiz.

+0

APK'nın beklediğiniz PB dosyalarını içerip içermediğini kontrol edebilir misiniz? Ayrıca, bazı hata izleme ve daha fazla ayrıntı elde etmek için yardımcı olabilir. –

cevap

0

DecodeJpeg, çekirdeğin bir parçası olarak desteklenmediğinden, önce grafiğin dışına çıkarmanız gerekir. Bu dosyada

bazel build tensorflow/python/tools:strip_unused && \ 
bazel-bin/tensorflow/python/tools/strip_unused \ 
--input_graph=your_retrained_graph.pb \ 
--output_graph=stripped_graph.pb \ 
--input_node_names=Mul \ 
--output_node_names=final_result \ 
--input_binary=true 

Değişim birkaç parametre

/tensorflow/examples/android/src/org/tensorflow/demo/TensorFlowImageListener.java

giriş büyüklükleri 299, değil 224 olması gerekiyor. Ayrıca 128.

INPUT_NAME için ortalama ve std değerlerini hem değiştirmeniz gerekir için "Mul: 0", OUTPUT_NAME için "final_result: 0"

sonra hangi apk derlemek mümkün olacak.

İyi Şanslar

+0

Merhaba Rohit, TensorFlow 0.10.0rc0 kullanıyor musunuz? Ben bazel-bin/tensorflow/python/tools/strip_unused' çalıştırdığımda hata tensorflow.python.tools 'import.unused_lib importError'dan alırsınız: Tools' adlı bir modül yok .. Teşekkürler, Jeff –

+0

r0.9'dayım. "tools: strip_unused" değil "tools/strip_unused", çalışma alanınızdan terminalde tüm komutu bir kerede tümüyle çalıştırın ve .pb dosyanızın çalışma alanınızda bulunduğundan emin olun. – Rohit