如何从 google colab 加载保存的 h5 文件并在 PC 中使用

Question

我使用预训练的 CNN 训练了一个模型，并将其以 h5 格式保存在 Colab 上。对于我的项目，我需要从四个摄像头捕获图像，并使用经过训练的模型进行图像分类。 Colab 似乎无法连接到四个摄像头，我正在考虑用我的电脑来连接。但是，我应该如何使用位于 Colab 上的 h5.file？我是编程和机器学习的新手。对不起，如果我问了一个愚蠢的问题......谢谢你

Answer 1

如果问题是关于从colab 下载模型，只需右键单击并从最左侧的文件浏览器窗格中选择下载即可。

如果问题旨在从模型中获取预测。一个常见的模板如下

from tensorflow.keras.models import load_model
model=load_model("model.h5")
prediction=model.predict(your_input)
best_pred=np.argmax(prediction)

您也可以通过mounting驱动器将模型直接保存到您的谷歌驱动器。