Fork me on GitHub

神经网络模型的可视化

我们通常接触到一个新的网络模型,首先需要了解其网络结构(拓扑),这样才能帮我们进一步了解该网络的构成。针对caffe框架训练的模型,我们可以通过在线网站Netscope可视化其结构。不过该网站仅支持prototxt格式描述的神经网络结构。

我再向大家推荐一个简易工具Netron,可离线使用,支持当前主流框架训练的模型,例如Mxnet,Tensorflow,pytorch,darknet。

您的鼓励是我持之以恒的动力