2016-02-06 15 views
8

Sieć początkowa w modelu Android Tensorflow Demo przechowuje model jako plik protobuf (tensorflow_inception_graph.pb). Chciałbym zastąpić tę sieć inną.Przekształcanie wyszkolonego modelu Tensorflow na protobuf

Czy istnieje dokumentacja lub przykład gdzie przekonwertować wyszkoloną sieć TensorFlow w języku Python na .pb? Jestem świadomy tego, że TensorFlow ma Saver, ale wydaje się, że to służy do oszczędzania stanu treningu pośredniego. Nie wiesz, jak to działa, jeśli model jest już przeszkolony.

+0

jeśli jedna z wymienionych odpowiedzi jest prawidłowa lub znalazłeś inne rozwiązanie i aktualizacja byłaby miła – BraveHeart

Odpowiedz

4

tf.Graph.as_graph_def() pobiera szeregowany wykres. Następnie po prostu zapisz go w pliku.

Powiązane problemy