在pycharm上加载经过训练的Tensorflow Saved模型时出错 ValueError:以10为基数的int的无效文字:'class_name'

问题描述

上下文: 我已经训练了一个包含图像分类器的模型。这个模型包含了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中起作用?
-如何解决此问题,以便在我的应用中加载和使用经过训练的模型?

解决方法

  1. 适用于 colab 但不适用于 IDE,因为您在两个位置安装了不同版本的 tenserflow
  2. 将 Tensorflow 版本从 2.0.0 更改为 2.3.1 可解决此问题。

相关问答

依赖报错 idea导入项目后依赖报错,解决方案:https://blog....
错误1:代码生成器依赖和mybatis依赖冲突 启动项目时报错如下...
错误1:gradle项目控制台输出为乱码 # 解决方案:https://bl...
错误还原:在查询的过程中,传入的workType为0时,该条件不起...
报错如下,gcc版本太低 ^ server.c:5346:31: 错误:‘struct...