caffe（7） 训练自己的CNN网络—基于自己建立的数据集

首先参照博客http://www.cnblogs.com/denny402/p/5083300.html和http://blog.youkuaiyun.com/u012029332/article/details/51258617，使用博主的数据训练，前面都很顺利，到最后运行时报内存错误，可能的解决方案为：减小batch_size或者增大内存。接下来使用自己的数据训练caffe网络，步骤一一记录下来。

使用第二个博客的步骤训练，时间大概为半个小时，结果为：loss = 0.33919， accuracy = 0.93

okay，现在开始训练自己的网络。

数据集为网上找的数据，共三类人脸、指纹和虹膜，每类500张，共1500张。其中1200张作为训练集，300张作为测试集。

第一步：首先生成图像的列表文件

我的三类文件分别在三个文件夹中，每一个文件夹下面还有子文件夹，采用脚本文件生成文件列表，注意標籤從0開始，create_filelist.sh如下：

#!/usr/bin/env sh
DATA=data/song/
MY=examples/myfile2

echo "Create train.txt..."
rm -rf $MY/train.txt

find $DATA/CASIA_Lamp_500 -name *.jpg | sed "s/$/ 0/">>$MY/train.txt
find $DATA/FVC2004_fingerprints_500 -name *.tif | sed "s/$/ 1/">>$MY