在TensorFlow中使用多个模型

时间:2017-06-02 11:01:51

标签: tensorflow

我有一个模型可以检测多达10位数字,我使用相同的图形结构来检测4个字母。这意味着第一个模型有10个类,第二个模型有4个类。不幸的是,我在训练时使用了相同的变量名称。 现在我试图创建一个代码来使用这两个模型来检测数字以及代码中给出的字母。

  graph_1 = tf.Graph() 
  with graph_1.as_default() as g_1:
    ...graph_1 content.....

  graph_2 = tf.Graph() 
  with graph_2.as_default() as g_2:
    ...graph_2 content.....

  sess_1 = tf.InteractiveSession(graph = g_1)
  sess_1.as_default()
  ....restoring model....

  sess_2 = tf.InteractiveSession(graph = g_2)
  sess_2.as_default()
  ....restoring model....
  .
  .
  .
  sess_1.run(prob, feed_dict)        
  sess_2.run(prob, feed_dict)      

但是现在,我知道这不是正确的接近方式。如果有人知道,请帮助我安排此代码或分享一些有用的链接。 提前谢谢。

0 个答案:

没有答案