Caffe学习之模型可视化

最新推荐文章于 2021-05-10 21:17:49 发布

原创最新推荐文章于 2021-05-10 21:17:49 发布 · 400 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #可视化工具 #网络 #结构 #caffe

caffe相关专栏收录该内容

2 篇文章

订阅专栏

本文介绍如何使用Python环境下的Caffe工具实现网络结构及权值的可视化。通过编译pycaffe，可以绘制出如caffenet、alexnet等网络的结构图，并提供了具体的命令行操作步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1、网络可视化

准备Python环境
因为caffe是使用基于Python的网络结构可视化工具，所以需要Python环境，即编译pycaffe后就可使用。

# cd ${CAFFE_ROOT}
# make clean
# make -j16
# make pycaffe -j16
顺便编译一下MATLAB的环境
# make matcaffe -j16

绘制网络结构图

# cd ${CAFFE_ROOT}/python
# python draw_net.py ../models/bvlc_refence_caffenet/train_val.prototxt caffenet.png
# python draw_net.py ../models/bvlc_alexnet/train_val.prototxt alexnet.png
# python draw_net.py ../models/bvlc_googlenet/train_val.prototxt googlenet.png
也可以根据自己设计的网络进行可视化。