深度神经网络结构可视化(Pytorch、Tensorflow、Keras、Caffe、MXNet、PaddlePaddle等主流框架)
本文代码存放于github上参考可视化深度神经网络结构 本文我们使用三种方式对神经网络结构进行可视化,其中Netron工具适用范围最广,基本可用于大部分主流框架;hiddenlayer工具包适用于Pytorch、Tensorflow、Keras框架;而pytorchviz工具包只适用于Pytorch。(链接都指向github库) 本文以Pytorch框架下的resnet34为例进行可视化,只是