Tensor Fluss kann nicht geladen werden benutzerdefinierte protobuf Dateien in

Question

Ich habe gerade eine protobuf Datei erstellt (.pb-Datei) für meine eigenen benutzerdefinierten Bilder eineine TensorFlow tutorial verwenden.

Aber wenn ich die gleiche Datei in den Ordner Vermögenswerte in tensorflow/examples/android/assets ersetzte und versuche, ein APK zu erstellen und zu generieren, wird das APK generiert, aber wenn ich das APK in einem Android-Gerät ausführen, stürzt das APK ab.

Wenn ich die classify_image von Python laufen, es gibt mir gute Ergebnisse zu erzielen.

Schätzen Sie jede Hilfe.

Answer 1

Da DecodeJpeg wird nicht als Teil des Kerns unterstützt, werden Sie es zuerst aus dem Diagramm strippen müssen.

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 

ändern einige Parameter in dieser Datei

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

Die Eingangsgrößen benötigen 299, nicht 224 sein. Sie werden auch die mittleren und Standardwerte beide auf 128

INPUT_NAME auf "Mul: 0" ändern müssen, output_name zu "final_result: 0"

, nach denen Sie die apk kompilieren können.

Viel Glück