模型可视化
关于模型的可视化方面,最近找到一个比较方便的软件,netron,可以比较容易实现可视化。
最终的效果如下所示:
需要注意的是,他现在对PyTorch的.pth支持不是很好,可以保存为.onnx进行绘制,可以使用如下方式进行保存。
- import torch.onnx
- # 输入测试数据
- input_data = Variable(torch.randn(1,1,64,64))
- torch.onnx.export(resNet, input_data, "ResNet.onnx", export_params=True, verbose=True, training=False)
把下载的链接放在最后了。
参考链接
- 微信公众号
- 关注微信公众号
- QQ群
- 我们的QQ群号
评论