如何在脚本中恢复两个tensorflow模型

时间:2017-03-08 02:32:29

标签: tensorflow

我写了两个分别包含分段和分类的脚本。这两个模型都经过训练,我想用它们来做预测。分类的输入基于分段的输出。如

from model1 import segmentation
from model2 import classification

images = segmentation(img)
pred = classification(images)

问题是这两个句子可以单独运行但不能按顺序运行。有关train.saver的问题将会发生。

我该如何解决这个问题?

0 个答案:

没有答案