使用张量流队列将输入传递给预训练的初始模型

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.FIFO Feed，而不是feed dict。我怎么能这样做？

Answer 1

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