直接使用caffe中的画图功能,会遇到问题: ImportError: No module named pydot
,说明缺少pydot库
直接安装pydot:
pip install pydot
重新测试画图
~/ml/caffe/python$ python draw_net.py \
../examples/mnist/lenet_train_test.prototxt lenet5.png
Drawing net to lenet5.png
注: 如果没有安装graphviz, 会出现Exception: "dot" not found in path.
的问题
安装graphviz不要用pip install安装,否则还是会找不到可执行程序
安装:
sudo apt-get insall graphviz
转载自:
pydot 安装 用于caffe画图