深度学习模型可视化-Netron

  • A+
所属分类:深度学习
摘要记录一下一个好用的神经网络可视化的软件Netron,并记录一下Pytorch中模型的保存。

模型可视化

关于模型的可视化方面,最近找到一个比较方便的软件,netron,可以比较容易实现可视化。

最终的效果如下所示:

深度学习模型可视化-Netron

需要注意的是,他现在对PyTorch的.pth支持不是很好,可以保存为.onnx进行绘制,可以使用如下方式进行保存。

  1. import torch.onnx
  2. # 输入测试数据
  3. input_data = Variable(torch.randn(1,1,64,64))
  4. torch.onnx.export(resNet, input_data, "ResNet.onnx", export_params=True, verbose=True, training=False)

把下载的链接放在最后了。

参考链接

github仓库--netron

深度学习模型可视化-Netron
  • 微信公众号
  • 关注微信公众号
  • weinxin
  • QQ群
  • 我们的QQ群号
  • weinxin
王 茂南

发表评论

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen: