[caffe]将自己的图片数据转换db格式

我们手中有的一般都是图片数据，jpg、bmp格式等，但caffe常使用的数据是db格式（leveldb/lmdb），因此首先我们要将自己数据转换成caffe可运行的格式文件。别捉鸡，caffe有给我们提供工具。根目录的tools文件下convert_imageset.cpp，经编译生成可执行文件，存放于./build/tools/文件，这个文件可帮我们将图片文件转换成db格式。

以分类任务为例，需要4个文件，train、train.txt、val、val.txt，train和val分别存放训练数据和测试数据，而两个txt文件分别是其文件的lable。

我们可以创建一个sh脚本来执行所有的操作。（以下操作均在/path to/caffe-master/路径下）

sudo vim data/mydata/create_lmdb.sh（我的数据放在data文件下的mydata目录下）

创建好脚本文件后即在里面写入代码：

#！ /usr/bin/env sh

DATA = data/mydata（此为你自己数据的存放路径，包含4个文件）

echo " create img_train.lmdb"

rm  -rf  /path to/caffe-master/examples/your file/img_train.lmdb （此处先把你要存放lmdb文件目录下原来的lmdb文件删除）

build/toold/convert_imageset   --shuffle=true   \（shuffle为是否打乱数据的顺序，默认为false）

--resize_height=28   --resize_width=28   \（resize图片的尺寸）

/path to/caffe-master/data/mydata/train/\ （训练数据的存放路径）

$DATA/train.txt\ （训练数据lable文件的存放路径）

/path to/caffe-master/examples/your file/img_train.lmdb（此处为生成的lmdb文件的存放路径）

echo " create img_val.lmdb"

rm  -rf  /path to/caffe-master/examples/your file/img_val.lmdb

build/toold/convert_imageset   --shuffle=true   \

--resize_height=28   --resize_width=28   \

/path to/caffe-master/data/mydata/val/\

$DATA/val.txt\

/path to/caffe-master/examples/your file/img_val.lmdb

echo "All Done!"

保存退出，然后执行该sh脚本文件即可生成lmdb文件。

sudo sh data/mydata/create_lmdb.sh

Prolem：

1、刚开始在sh脚本里只写了一段代码，即生成img_train.lmdb再修改脚本生成img_val.lmdb，但是在执行的时候出问题了，没有细探原因，也可能是我自己当时没有完全改过来吧。

Reference：