如何在 Pycharm 中使用从 google colab 导出的模型答案

【问题标题】：How to use an exported model from google colab in Pycharm如何在 Pycharm 中使用从 google colab 导出的模型
【发布时间】：2020-06-14 09:38:03
【问题描述】：

我有一个以 .h5 (HDF5) 格式训练和导出的 LSTM Keras Tensorflow 模型。我的本地机器不支持 keras tensorflow。我试过安装。但不起作用。

因此，我使用了 google colabs 并导出了模型。

我想知道，如何在 pycharm 中使用导出的模型

编辑：我刚刚在我的机器上安装了 tensorflow

提前致谢

【问题讨论】：

【解决方案1】：

接受的答案是正确的，但它错过了您首先需要安装“/content/drive”

from google.colab import drive
drive.mount('/content/drive')

然后就可以保存模型的权重了：

model.save_weights('my_model_weights.h5')

..甚至保存整个模型：

model.save('my_model.h5')

完成后，使用以下命令断开挂载点：

drive.flush_and_unmount()

【讨论】：

【解决方案2】：

找到答案了：

我已经导出模型如下

model.save('/content/drive/My Drive/Colab Notebooks/model.h5')

然后我下载了文件并保存在我的其他代码所在的文件夹中。我已经安装了张量流。

接下来我加载代码并使用保存的模型进行预测，如下所示。

import keras
model=keras.models.load_model('/content/drive/My Drive/Colab Notebooks/model.h5')
model.predict(instace)

【讨论】：

【解决方案3】：

你仍然需要 keras 和 tensorflow 来使用模型。

【讨论】：