初试caffe训练,这里不是跑的caffe里自带的例子,是训练测试自己的数据集,欢迎指正~ps:这边就不介绍caffe的安装配置了,直接进入主题啦~
1大量样本的准备
既然要做训练的话没有大量的图片样本是不行滴,这个阶段耗费很久的时间,说到这里要吐血了,三个人,对采集的视频中的每帧图片进行截图,不夸张的说总共截的图片大概有近百万张,当然啦还要对图片进行筛选不好的样本不要,否则会影响测试效果!我们一共做了四组实验,这里挑一个简单一组实验进行讲解。
有四类图片,每类都有大量的图片
介绍一种简易得到标签的方法,在每个文件夹下可建立一个批处理命令
运行getall.bat可得到
用notepad打开使用替换将jpg替换成jpg 1简易的标签就好了啊
2训练集和测试集
这组实验我们的样本总数为250465,训练集170318,测试集80147。
针对四类我们分别挑选了一些图片作为训练集,剩下的作为测试集。四个类所有训练的图片放在一起,所有测试的图片放在一起
原本四个类我们是从1-4的,后来我们将其批量改成了0-3