如何解决在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中起作用?
-如何解决此问题,以便在我的应用中加载和使用经过训练的模型?
解决方法
- 适用于 colab 但不适用于 IDE,因为您在两个位置安装了不同版本的 tenserflow
- 将 Tensorflow 版本从 2.0.0 更改为 2.3.1 可解决此问题。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。