huachao1001 / cnngraph

A tool that automatically extracts network structures from Tensorflow model files

• 👋 Hi, I’m @huachao1001
• 👀 I’m interested in ...
• 🌱 I’m currently learning ...
• 💞️ I’m looking to collaborate on ...
• 📫 How to reach me ...

cnngraph's Issues

Need change some code if user use python2.7

GraphBuilder.py
`
+# -- coding:utf-8 --
import io
import sys
reload(sys)
sys.setdefaultencoding('utf-8')

def build(self):
change all "open('html/NodeObj.js','r',encoding='utf-8')" as r: with
io.open('html/NodeObj.js','r',encoding='utf-8') as r:
`
add code to MergeLayers.py
+# -- coding:utf-8 --

read_graph.py
replace all list.copy with for-statement

Another problem is that if NN graph is too complicated, the HTML file is seem awkward.

输入为什么非要手动输入input/output的tensor啊

醉了

不知道输入和输出tensor的名称

不知道输入和输出tensor的名称，应该怎么做呢？

AttributeError: 'NoneType' object has no attribute 'outputs'

在调用merge_layers函数中出现了这个， 请问如何解决呀

关于工具的使用问题

请问在tensorflow2.x的模型中使用报错，要怎么修改

作者您好，看代码的时间，应该是适配tf1.x的，但我现在有个tf2.x的（pb和ckpt），想打印它的结构,打印pb模型的时候报错：
Traceback (most recent call last):
File "read_graph.py", line 189, in
read_graph(model_path,input_names,output_name,html_dst)
File "read_graph.py", line 177, in read_graph
ops = read_graph_from_pb( model_path ,input_names,output_name)
File "read_graph.py", line 25, in read_graph_from_pb
tf.reset_default_graph()
AttributeError: module 'tensorflow' has no attribute 'reset_default_graph'

输入tensor中有一个维度是可变的怎么办

html error

when i run html, problem happened as follow:
Uncaught TypeError: Cannot read property 'addNode' of undefined
at MergedGraph.mergeOps (dst.html:475)
at MergedGraph.genSvg (dst.html:491)
at changeView (dst.html:555)
at dst.html:560

关于html显示为空的问题

您好，为什么有些模型生成html文件后能正常显示；而有些模型生成的html文件打开后没有内容，程序运行中也没报错，正确生成了html文件。

Help for error!

hello，
I got this error when using your code. I really appreciate you can help me with this error. Thank you!
error:
G[op_name_to_index[inp.op.name]].append(i)
AttributeError: 'NoneType' object has no attribute 'name'