Caffe-Windows下画loss与accuracy曲线

本篇博客详细介绍了如何在Windows环境下使用Caffe提供的工具,结合训练日志生成loss和accuracy曲线。内容包括修改plot_training_log.py文件,如创建field_index、解析日志并保存csv文件,以及load_data函数的调整。通过这些步骤,读者可以生成训练和测试的损失及准确率图表。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

请注意如果.py文件中有中文,请在文件开头加上# -*- coding: utf-8 -*-,否则会报错yntaxError:Non-ASCII character '\xe6' in file,文章最后原博主给的下载代码中没有这句话,需要大家手动加上。

另外代码是Python2.7版本的,如果是Python3版本的,运行会出错,解决方案请参考http://blog.youkuaiyun.com/zyx19950825/article/details/61919038


本篇博客主要讲述怎样在Windows下利用Caffe提供的脚本程序和Caffe训练日志画loss曲线与accuracy曲线。如果你是在Linux下使用Caffe可以参考这篇博客:http://blog.youkuaiyun.com/fx409494616/article/details/53197209?ref=myread

如果你还没有Caffe训练日志,请参考上一篇博客http://blog.youkuaiyun.com/fuwenyan/article/details/62418108,生成自己的训练日志。


最终结果:

在生成log的文件夹中要放如下几个文件:



1、修改tools/extra/plot_training_log.py,这里面需要修改的东西太多了,我们分步讲解,可能代码优点乱,大家不要介意。

1.1、生成*****log.test,*****log.train两个文件

方法一:利用tools/extra/parse_log.py文件

[python]  view plain   copy
  在CODE上查看代码片 派生到我的代码片
  1. python parse_log.py ****.log save_path  
第一个参数:我们的训练日志,后缀名必须是".log",其实这也不是必须的,我们可以修改plot_training_log.py中子函数

[python]  view plain   copy
  在CODE上查看代码片 派生到我的代码片
  1. def get_log_file_suffix():  
  2.     return '.log'#可以返回其他后缀名  
第二个参数:保存路径,执行上述命令后会生成两个文件****.log.test,****.log.train。

方法二:将生成这两个文件集成到plot_training_log.py中。我们首先看一下两个plot_training_log.py文件中的子函数

[python]  view plain   copy
  在CODE上查看代码片 派生到我的代码片
  1. def get_log_parsing_script():  
  2.     dirname = os.path.dirname(os.path.abspath(inspect.getfile(  
  3.         inspect.currentframe())))  
  4.     return dirname + '/parse_log.sh'  
返回的是parse_log.sh脚本的路径,看来要调用这个脚本,但是我们知道在Windows下是无法使用shell脚本的。所以我们需要修改调用这个shell脚本的地方。就在下面这个子函数

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值