Wie die Wiederherstellung Tensorflow Modell aus .pb-Datei in python?

Ich habe eine tensorflow .pb-Datei, die ich gerne laden in python DNN, wiederherstellen der Grafik und Holen Sie sich die Vorhersagen. Ich Tue dies, um zu testen, ob die .pb-Datei erstellt, können die Vorhersagen ähnlich wie die normalen Sparer.save () - Modell.

Mein grundlegende problem ist, bin ein ganz anderer Wert von Vorhersagen, wenn ich Sie auf Android mit den oben genannten .pb-Datei

Meine .pb-Datei erstellen code:

frozen_graph = tf.graph_util.convert_variables_to_constants(
        session,
        session.graph_def,
        ['outputLayer/Softmax']
    )
with open('frozen_model.pb', 'wb') as f:
  f.write(frozen_graph.SerializeToString())

Also habe ich zwei große Bedenken:

  1. Wie kann ich laden Sie die oben genannten .pb-Datei zu python Tensorflow Modell ?
  2. Warum bin ich immer ganz unterschiedliche Werte der Vorhersage in python und android ?
InformationsquelleAutor vizsatiz | 2018-05-31
  1. 22

    Folgende code liest die Modell und drucken Sie die Namen der Knoten im graph.

    import tensorflow as tf
from tensorflow.python.platform import gfile
GRAPH_PB_PATH = './frozen_model.pb'
with tf.Session() as sess:
   print("load graph")
   with gfile.FastGFile(GRAPH_PB_PATH,'rb') as f:
       graph_def = tf.GraphDef()
   graph_def.ParseFromString(f.read())
   sess.graph.as_default()
   tf.import_graph_def(graph_def, name='')
   graph_nodes=[n for n in graph_def.node]
   names = []
   for t in graph_nodes:
      names.append(t.name)
   print(names)

    Sie sind zum einfrieren der Grafik richtig, das ist, warum Sie immer verschiedene Ergebnisse grundsätzlich GEWICHTE sind nicht immer gespeichert in Ihrem Modell. Sie können die freeze_graph.py (link) für eine korrekt gespeichert graph.

    • Was ist sess.graph.as_default() tun?
    • ich bekomme auf graph_def.ParseFromString(f.read()) DecodeError: Fehler beim Parsen der Nachricht
    • Verwenden tf.gfile.GFile statt gfile.FastGFile im Jahr 2019
    • Sie Lesen Datei-handler f aus with Umfang
    • Wenn ich versuche die Anzeige von node-Namen mit Ihrem Programm und facenet Modell , es ist mir dieser Fehler ValueError: Input 0 of node InceptionResnetV1/Conv2d_1a_3x3/BatchNorm/cond/Switch was passed float from phase_train:0 incompatible with expected bool. haben Sie eine Idee, warum es passiert ist ? Dank
    • Sie sind wahrscheinlich vorbei an falschen Datentyp. Es erwartet bool, aber es ist immer float.
    • Eigentlich habe ich nicht übergeben, nichts. Wenn Sie Zeit haben, können Sie bitte schauen Sie unter dieses Problem ?

    InformationsquelleAutor sahu
Schreibe einen Kommentar