使用caffe的可视化程序,绘制网络模型图
命令如下:
cd caffe
sudo 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