caffe 绘制网络模型图

来源:互联网 发布:mac os 显示内容放大 编辑:程序博客网 时间:2024/04/29 17:02

使用caffe的可视化程序,绘制网络模型图
命令如下:

cd caffesudo python ./python/draw_net.py ./examples/mnist/lenet_train_test.prototxt ./examples/mnist/lenet_train_test.jpeg

错误一:ImportError:No module named skimage.io
解决办法:安装 skimage.io模块

pip install skimage.io

错误二:ImportError:No module named pydot
解决办法:先安装graphviz,否则会出现找不到路径的问题

sudo apt-get graphviz

然后安装pydot

pip install pydot
原创粉丝点击