caffe绘制网络结构图

最新推荐文章于 2019-11-09 00:18:25 发布

VIP文章 Aiben_

最新推荐文章于 2019-11-09 00:18:25 发布

阅读量273

点赞数 1

分类专栏： Deep learning 文章标签： caffe pydot python

本文链接：https://blog.csdn.net/qq_41062824/article/details/83055368

版权

运行脚本文件draw_caffe.bat

python C:/caffe/python/draw_net.py  C:/caffe/examples/mnist/lenet_train_test.prototxt D:/lenetmap.png --rankdir=LR
pause

参数说明：
第一个参数：网络定义文件
第二个参数：生成图片的保存名称和路径
第三个参数：表示网络绘制方向，有四种，LR 从左到右,RL 从右到左, TB 从上到下 ,BT 从下到上。

双击后运行脚本，结果提示少库。

ImportError：No module named pydot

用pip进行安装

pip install pydotplus

注意：安装pydot可能会有AttributeError: 'module' object has no attribute 'graphviz' 错误

生成网络结构图片

Aiben_

关注