我通过Google Colab训练了我的CNN(VGG)并生成了.h5文件。现在的问题是,我可以在Google Colab上成功预测输出,但当我下载这个训练好的.h5模型文件并尝试在我的笔记本电脑上预测输出时,加载模型时会出现错误。
这是我的代码:
import tensorflow as tffrom tensorflow import kerasimport h5py# 初始化loaded_model = keras.models.load_model('./train_personCount_model.h5')
这是错误信息:
ValueError: Unknown initializer: GlorotUniform
回答:
我遇到了同样的问题。在将:
from tensorflow import keras
更改为:
import keras
之后,生活又变得值得继续了。