问题描述
上下文: 我已经训练了一个包含图像分类器的模型。这个模型包含了TF CNN,并且已经在网上进行了训练(使用google colaboratory),然后将其作为tf.savemodel对象下载了。我想在PyQt5桌面应用上部署经过训练的模型。
问题: 当我尝试在PyCharm(2020.1.3)的代码中加载经过训练的模型时,出现以下错误:
ValueError: invalid literal for int() with base 10: 'class_name'
在处理上述异常期间,发生了另一个异常:
ValueError: Error converting shape to a TensorShape: invalid literal
for int() with base 10: 'class_name'.
用于加载模型的代码:
model1=tf.keras.models.load_model(r'C:\Users\user\Desktop\UNI\L3\saved_models\model7')
侧面注意:当我在colab上运行同一行时,模型已成功加载到笔记本中。
问题:
-错误是什么意思?
-为什么它在colab上而不在我的IDE中起作用?
-如何解决此问题,以便在我的应用中加载和使用经过训练的模型?
解决方法
- 适用于 colab 但不适用于 IDE,因为您在两个位置安装了不同版本的 tenserflow
- 将 Tensorflow 版本从 2.0.0 更改为 2.3.1 可解决此问题。