使用PlotNeuralNet工具生成卷积神经网络的立体结构图

在写论文过程中我们需要一个清晰的网络结构图,而生成漂亮的立体图像能给我们的论文加分不少。生成的图像如图所示。
更多方法参考:22 款设计和可视化神经网络的工具
使用PlotNeuralNet工具生成卷积神经网络的立体结构图_第1张图片
具体实现过程如下:
1.下载PlotNeuralNet源代码链接
2.按照里面的README.md文件里的内容进行操作。
3.编写py文件。例子如下:

import sys
sys.path.append('../')
from pycore.tikzeng 

你可能感兴趣的:(pytorch,python,计算机视觉)