我们可以将在一台PC上训练的模型转移到另一台PC上吗

Question

我已经使用tenserflow和keras在MAC上训练了cnn模型，是否可以将此训练后的模型移到另一台装有Windows的PC上？如果是，那么我们可以在没有安装tenserflow的情况下，使用训练有素的模型进行预测吗？

Answer 1

您可以使用save()方法保存训练好的模型。要再次加载训练有素的模型，可以使用库提供的功能：

from keras.models import load_model

model.save('my_model.h5')  # creates a HDF5 file 'my_model.h5'
del model  # deletes the existing model

# returns a compiled model
# identical to the previous one
model = load_model('my_model.h5')

如文档所示：Documentation

对于在未安装keras或tensorflow的情况下执行模型的问题，简短的回答是“否”。长答案是肯定的，但是您将不得不使用任何其他库或自己实现一个函数，该函数采用经过模型训练的权重和体系结构，并进行通用前馈以返回结果。