在 pycharm 上加载经过训练的张量流保存模型时出错。ValueError: int() 的文字无效,底数为 10: 'class_name'



上下文:我训练了一个由图像分类器组成的模型。这个模型包含在一个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中无效
-如何修复此问题,以便在我的应用程序中加载和使用经过训练的模型?

  1. 在colab上工作,但在IDE上不工作,因为在这两个位置都安装了不同版本的tenserflow
  2. 将Tensorflow版本从2.0.0更改为2.3.1解决了这个问题

检查当前系统中TF的版本,您可能已经用2.1进行了训练,并试图用2.3版本的加载模型

相关内容

最新更新