首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >google.protobuf.message.DecodeError:使用'tensorflow.GraphDef‘类型的错误解析消息

google.protobuf.message.DecodeError:使用'tensorflow.GraphDef‘类型的错误解析消息
EN

Stack Overflow用户
提问于 2022-03-11 13:16:28
回答 1查看 4.6K关注 0票数 3

我训练模型并保存它,现在我试图加载,但无法做到。我在前一篇文章中也看到过,但是一些参考链接不起作用,或者我尝试了一些东西,仍然无法解决这个问题。

代码片段:

代码语言:javascript
复制
#load model

with tf.io.gfile.GFile(args.model, "rb") as f:
    graph_def = tf.compat.v1.GraphDef()
    graph_def.ParseFromString(f.read())

# with tf.Graph().as_default() as graph:
generated_image_1, generated_image_2, generated_image_3, = tf.graph_util.import_graph_def(
        graph_def, 
        input_map={'input_image' : input_tensor, 'short_edge_1' : short_edge_1, 'short_edge_2' : short_edge_2, 'short_edge_3' : short_edge_3}, 
        return_elements=['style_subnet/conv-block/resize_conv_1/output:0', 'enhance_subnet/resize_conv_1/output:0', 'refine_subnet/resize_conv_1/output:0'],  
        producer_op_list=None
    )

误差

代码语言:javascript
复制
Traceback (most recent call last):

  File "stylize.py", line 97, in <module>
    main()
  File "stylize.py", line 57, in main
    graph_def.ParseFromString(f.read())
google.protobuf.message.DecodeError: Error parsing message with type 'tensorflow.GraphDef'

注意:如果需要更多有关这方面的信息,一定要在这里添加。让我知道

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2022-03-19 17:14:00

BG:我在测试代码时出错了。在我的例子中,它是通过freeze.py和训练文件中的一些修改来解决的。在搜索查询时,我找到了一些其他有用的链接。Link 1

Link 2

票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/71439124

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档