Używam niestandardowego zestawu obrazów do trenowania sieci neuronowej przy użyciu interfejsu Tensorflow API. Po udanym treningu otrzymuję pliki punktów kontrolnych zawierające wartości różnych zmiennych treningowych. Teraz chcę uzyskać model wnioskowania z tych plików punktów kontrolnych, znalazłem to, script, które to robi, co mogę następnie wykorzystać do generowania obrazów w deepdream, jak wyjaśniono w tym samouczku. Problemem jest to, gdy załadować mojego modelu przy użyciu:Jak korzystać z modeli wnioskowania Tensorflow do generowania obrazów podobnych do Deepdreama
import tensorflow as tf
model_fn = 'export'
graph = tf.Graph()
sess = tf.InteractiveSession(graph=graph)
with tf.gfile.FastGFile(model_fn, 'rb') as f:
graph_def = tf.GraphDef()
graph_def.ParseFromString(f.read())
t_input = tf.placeholder(np.float32, name='input')
imagenet_mean = 117.0
t_preprocessed = tf.expand_dims(t_input-imagenet_mean, 0)
tf.import_graph_def(graph_def, {'input':t_preprocessed})
otrzymuję ten błąd:
graph_def.ParseFromString(f.read())
self.MergeFromString(serialized)
raise message_mod.DecodeError('Unexpected end-group tag.') google.protobuf.message.DecodeError: Unexpected end-group tag.
Skrypt oczekiwać plik bufora protokół, nie jestem pewien, że script używam do generowania modeli wnioskowania daje mi pliki bufora proto, czy nie.
Czy ktoś może zasugerować, co robię źle, czy jest lepszy sposób na osiągnięcie tego. Po prostu chcę konwertować pliki punktów kontrolnych generowane przez tensor na bufor proto.
Dzięki