6

Tensorflow API kullanarak bir sinir ağını eğitmek için özel bir görüntü kümesi kullanıyorum. Başarılı bir eğitim sürecinden sonra farklı eğitim varyasyonlarını içeren bu kontrol noktası dosyalarını elde ederim. Şimdi bu denetim noktası dosyalarından bir çıkarım modeli almak istiyorum, bu script buldum, ki bu daha sonra bu öğreticide açıklandığı gibi derin resimler oluşturmak için kullanabilirsiniz. Ben çıkarım modellerini üretmek için kullanıyorum script emin değilim,Tensorflow çıkarım modellerini, görüntüler gibi derinlemesine oluşturmak için nasıl kullanılır?

graph_def.ParseFromString(f.read())

self.MergeFromString(serialized)

raise message_mod.DecodeError('Unexpected end-group tag.') google.protobuf.message.DecodeError: Unexpected end-group tag.

komut bir protokol tampon dosyası bekliyoruz:

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}) 

bu hatayı alıyorum: Sorun kullanıyorum Modelimin yüklediğinizde olduğunu bana proto tampon dosyaları veriyor ya da vermiyor.

Birisi neyi yanlış yaptığımı önerebilir veya bunu başarmanın daha iyi bir yolu olabilir mi? Ben sadece tensör tarafından oluşturulan kontrol noktası dosyalarını proto-buffer'a dönüştürmek istiyorum.

Teşekkür

cevap

1

sen ran senaryoya bağlantı bozuk olduğunu, ancak her durumda tavsiye edilen şey, bir kontrol noktasında bir çıkarsama modelini oluşturmak için denemek için değil, sizin eğitim sonunda kodunu gömmek için değil "SavedModel" dışa aktarma (değil kontrol noktası ile aynı şey) yayınlayacak program.

Lütfen özellikle [1] ve özellikle de "Kaydedilmiş Model Oluşturma" başlığına bakınız. Kayıtlı bir Modelin birden fazla dosya oluşturduğunu unutmayın, bunlardan biri gerçekten de bir protokol tamponudur (bu, doğrudan sizin sorunuza cevap verir); diğerleri değişken dosyalar ve (isteğe bağlı) varlık dosyalarıdır.

[1] https://www.tensorflow.org/programmers_guide/saved_model

İlgili konular