1.下载mnist数据库
// cd 到caffe根目录
$ ./data/mnist/get_mnist.sh
// 转换成lmdb数据格式
$ ./examples/mnist/create_mnist.sh
// 训练mnist模型
$ ./examples/mnist/train_lenet.sh
// 默认是GPU模式进行训练的
2.本地可视化lenet网络图
$ sudo -H pip install pydot //需要安装pydot,否则画图会提示报错
$ sudo apt-get install graphviz //否则会提示Exception: "dot" not found in path.
$ python python/draw_net.py xxx.prototxt xxx.png --rankdir=TB
3.在线可视化工具
http://ethereon.github.io/netscope/#/editor
- draw_net.py画图工具使用问题:
//问题现象如下
......
File "/home/andre/caffe/python/caffe/draw.py", line 94, in get_layer_label
layer.convolution_param.kernel_size[0] if len(layer.convolution_param.kernel_size._values) else 1,
AttributeError: 'google.protobuf.pyext._message.RepeatedScalarConta' object has no attribute '_values'
//python/caffe/draw.py 文件修改如下:
(layer.name,
separator,
layer.type,
separator,
+ layer.convolution_param.kernel_size[0] if len(layer.convolution_param.kernel_size) else 1,
- layer.convolution_param.kernel_size[0] if len(layer.convolution_param.kernel_size._values) else 1,
separator,
+ layer.convolution_param.stride[0] if len(layer.convolution_param.stride) else 1,
- layer.convolution_param.stride[0] if len(layer.convolution_param.stride._values) else 1,
separator,
- layer.convolution_param.pad[0] if len(layer.convolution_param.pad._values) else 0)
+ layer.convolution_param.pad[0] if len(layer.convolution_param.pad) else 0)