2016-02-06 20 views
8

La rete iniziale nello Android Tensorflow Demo memorizza il modello come file protobuf (tensorflow_inception_graph.pb). Mi piacerebbe sostituire questa rete con un'altra.Conversione del modello Tensorflow addestrato in protobuf

C'è documentazione o un esempio da qualche parte su come convertire una rete TensorFlow addestrata in Python su .pb? Sono a conoscenza di Saver di TensorFlow ma sembra essere usato per salvare lo stato di addestramento intermedio. Non sei sicuro di come funziona se il modello è già stato addestrato.

+0

se una delle risposte citate è corretta o hai trovato una soluzione diversa e l'aggiornamento qui sarebbe bello – BraveHeart

risposta

4

tf.Graph.as_graph_def() recupera il grafico serializzato. Quindi basta scriverlo su un file.

Problemi correlati