【发布时间】:2019-04-10 13:15:19
【问题描述】:
我通过 google colab 训练了我的 CNN (VGG) 并生成了 .h5 文件。现在的问题是,我可以通过 google colab 成功预测我的输出,但是当我下载该 .h5 训练模型文件并尝试在我的笔记本电脑上预测输出时,我在加载模型时遇到错误。
代码如下:
import tensorflow as tf
from tensorflow import keras
import h5py
# Initialization
loaded_model = keras.models.load_model('./train_personCount_model.h5')
还有错误:
ValueError: Unknown initializer: GlorotUniform
【问题讨论】:
-
可能是由于 google colab 和您本地机器的 Keras 版本不匹配造成的。
-
okie让我检查一下..!
-
或者也有可能是tf.keras和keras混合造成的(不一样)。
-
@today 也许你是对的..我会承认这是否有效,反之亦然。
-
@MatiasValdenegro 我在两个平台上都使用过 tf.keras(即 google colab 和我的笔记本电脑)
标签: python tensorflow machine-learning keras google-colaboratory