使用tensorflow队列将输入传递给预训练初始模型

Question

我使用张量流中的预训练初始模型进行图像识别。我想使用tf.FIFO将输入传递给初始图。你知道吗

我当前将值传递为

input = #numpy array of image
softmax_tensor = sess.graph.get_tensor_by_name('final_result:0')
predictions = session.run(softmax_tensor, {'Cast:0': input})

现在我想在值中使用tf.FIFOfeed，而不是feed dict。我该怎么做？你知道吗

Answer 1

最灵活的方法是使用variable sharing，将旧的权重加载到带有输入队列的新图中。只要确保变量名是一致的。通常，您可以修改旧的图形生成代码以使用新的输入。你知道吗