caffe:用自己的图像数据训练模型

2016.4.6Alexweek

本文参考“学习笔记 3 用自己的数据训练和测试“CaffeNet（薛开宇）”和caffe官网。

1.数据准备

在caffe/data下新建目录myself，并在myself里又新建两个目录train、val。

 

注意：图片的格式必须为.jpeg格式

train存放训练用的数据源；该目录下又两个目录bird(70张图)、cat(70张图)

 

 

 

val存放用于测试的数据源；bird和cat各20张图

 

在终端下切换到caffe/data/myself目录下，利用上面的数据源生成train.txt、val.txt、test.txt。

test.txt的内容和val.txt相同，只是没有后面的数字标识。

 

生成val.txt的命令：find  -name *.jpeg |grep -v train | cut -d/ -f3>val.txt