参考博文https://blog.youkuaiyun.com/qq_27923041/article/details/54139887
一、准备数据
找了500张图片来代替,分为大巴车、恐龙、大象、鲜花和马五个类,每个类100张。需要的同学,可以去这个网盘下载点击打开链接。
编号分别以3,4,5,6,7开头,各为一类。我从其中每类选出20张作为测试,其余80张作为训练。因此最终训练图片400张,测试图片100张,共5类。我将图片放在caffe根目录下的data文件下。即训练图片目录:data/re/train/,测试图片目录:data/re/test/
***所有图片皆为*.jpg格式,需要转换格式lmdb格式
二、转换为LMDB格式
首先在examples下面创建一个example1的文件夹,来用存放配置文件和脚本文件。然后编写一个脚本create_filelist.sh,用来生成train.txt和test.txt清单文件
#mkdir examples/example1
#vi examples/example1/create_filelist.sh
***上述命令是在caffe目录下执行的;若当前是根目录,可以cd caffe,或者mkdir /home/song/caffe/examples/example1
***如果用sudo mkdir examples/example1,创建的文件夹(文件也一样)图标右下方有个锁,后续编写修改保存文件,会提示无法保存。可能是该文件具有root权限,可以用命令去掉:sudo chown -R $user 文件夹路径 来去除。chown命令是改变文件所有者,-R即连同目录下的所有子目录或文件都同时改变文件所有者
***vi文本编辑器(vim是vi的高级版本)基本上有三种模式:一般模式,编辑模式,命令模式。vi 打开文件即进入一般模式,此时可删除/复制/粘贴,但是无法编辑文件内容;按下“i,I,o,O,a,A,r,R"中任何字母中的一个即进入编辑模式,此时左下角会出现insert的字样,可以对文本进行编辑,完成后按esc退出编辑模式,回到一般模式;输入”/,:,?“三个中的任何一个按钮,即进入命令模式(显示在最下方),可以读取/保存/退出。命令模式下”:wq“即保存并退出。
编辑create_filelist.sh文件,并写入如下代码,并保存
#!/usr/bin/env sh
DATA=data/re/
MY=examples/example1
echo "Create train.txt..."
rm -rf $MY/train.txt
for i in 3 4 5 6 7
do
find $DATA/train -name $i*.jpg | cut -d '/' -f3-4 | sed "s/$/