上下文:我训练了一个由图像分类器组成的模型。这个模型包含在一个TF-CNN中,并且已经过在线训练(与谷歌合作(,然后我将其作为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:UsersuserDesktopUNIL3saved_modelsmodel7')
旁注:当我在colab上运行同一行时,模型会成功加载到笔记本中。
问题:
-错误是什么意思
-为什么它在colab上有效,而在我的IDE中无效
-如何修复此问题,以便在我的应用程序中加载和使用经过训练的模型?
- 在colab上工作,但在IDE上不工作,因为在这两个位置都安装了不同版本的tenserflow
- 将Tensorflow版本从2.0.0更改为2.3.1解决了这个问题
检查当前系统中TF的版本,您可能已经用2.1进行了训练,并试图用2.3版本的加载模型