点击此处返回总目录

 

跟在windows上一样。

 

一、网络结构可视化

命令:python  python/draw_net.py

参数1:模型文件

参数2:输出的图像文件

参数3:图形绘制方向

 

 

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

 

 

 

 

 

 

 

 

相关文章:

  • 2021-11-26
  • 2021-07-16
  • 2021-06-18
  • 2021-08-22
  • 2021-09-08
  • 2021-08-14
  • 2021-06-22
猜你喜欢
  • 2021-04-17
  • 2021-08-05
  • 2022-12-23
  • 2021-09-18
  • 2021-07-13
  • 2022-01-28
  • 2022-01-22
相关资源
相似解决方案