前言
深度学习中,CNN模型可视化有助于我们学习、分析网络模型。目前接触到的可视化有:特征图可视化、网络模型结构可视化。本文为大家介绍的就是网络模型结构可视化的两款小工具。
可视化工具
本文主要介绍Netscope和ConvNetDraw。
Netscope
Netscope是一个在线编辑,用于可视化卷积神经网络模型结构的小工具。目前,支持Caffe的prototxt文件可视化。
支持自定义网络结构,可视化输出结构,只要按照prototxt的格式写即可。
可视化效果:
体验地址:https://cuijiahua.com/netscope/quickstart.html
ConvNetDraw
ConvNetDraw也是一个用于可视化卷积神经网络模型结构的小工具。与Netscope不同的是,ConvNetDraw是使用可视化命令进行可视化的,而且命令很简单比如:
可视化效果:
体验地址:https://cuijiahua.com/convdraw/
总结
两个工具都有些bug,对于前端可视化卷积神经网络感兴趣的朋友,可以去他们的Github开源项目看看,帮助完善就更赞了。
Netscope开源项目地址:https://github.com/ethereon/netscope
ConvNetDraw开源项目地址:https://github.com/cbovar/ConvNetDraw
来源:
https://cuijiahua.com/blog/2018/10/trick-8.html