tensorflow - 如何有效地可视化网络架构？

Question

是否有某种软件可以做到这一点？具体来说，我想可视化 Resnet18。除了自己画，没有别的办法了吗？这是我想看到的一个例子：

示例架构可视化

标签： tensorflowdeep-learningpytorchvisualizationcode-visualization

Answer 1

你可以使用这个：http ://alexlenail.me/NN-SVG/LeNet.html 。它允许您通过修改多个参数来可视化神经网络，最后让您将架构导出为 SVG 文件。您还可以在 3 种可视化样式之间进行选择，即 FCNN、LeNet 和 AlexNet。