小编yiy*_*ing的帖子

如何从TensorFlow .pb模型中获取权重格式?

我想重新组织tensorflow .pb模型的节点,所以我首先从GraphDef获取NodeDef,然后使用NodeDef.attr()来获取"Conv2D"的节点.我能得到的参数,如进展,填充,DATA_FORMAT,use_cudnn_on_gpu从ATTR,但不能获取的权重格式参数.我使用的语言是c ++.怎么弄它!谢谢!

c++ model tensorflow

8
推荐指数
1
解决办法
455
查看次数

标签 统计

c++ ×1

model ×1

tensorflow ×1