网络结构可视化

                                                                                                                                                  点击此处返回总目录

 

跟在windows上一样。

 

一、网络结构可视化

命令:python  python/draw_net.py

参数1:模型文件

参数2:输出的图像文件

参数3:图形绘制方向。有四种选项:BT、TB、LR、RL

 

 

python /home/emcas/liupc/caffe/python/draw_net.py /home/emcas/liupc/caffe/examples/mnist/lenet_train_test.prototxt aa.png --rankdir=BT

 

 

 

二、学习曲线可视化

命令: python tools/extra/plot_training_log.py

第一个参数0-7意义如下,比如0代表准确率和迭代次数,1代表准确率和时间等等。

第二个参数:是图片的保存路径。

第三个参数:log。

 

cd /home/emcas/liupc/caffe/tools/extra

mv plot_training_log.py.example plot_training_log.py

python plot_training_log.py,可以看到用法:

 

                  

 

//注意,plot_training_log.py调用了同目录下的parse_log.py程序。而parse_log.py程序会将log分成两个训练集的log和测试集的log。

plot_training_log.py再根据这两个log中的数据画图。我们如果嫌plot_training_log.py画的不好看,可以先用parse_log.py生成两个log,再用matplotlib自己画图。

 

 

 

1. 当训练的时候,重定向一下,生成log文件。

cd /home/emcas/liupc/caffe/examples/mnist

caffe train -solver lenet_solver.prototxt 2>&1 | tee a.log

 

2. 执行画图的命令

python /home/emcas/liupc/caffe/tools/extra/plot_training_log.py 6 aaa.png a.log

 

 

 

 

 

 

 

 

你可能感兴趣的:(17,深度学习)