如何在Pycharm中使用从谷歌colab导出的模型

Question

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

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

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

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

提前感谢

Answer 1

找到答案：

我导出的模型如下

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

然后我下载了这个文件并保存在其他代码所在的文件夹中。我安装了tensorflow。

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

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

Answer 2

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

Answer 3

接受的答案是正确的，但它忽略了您首先需要装载">/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()