Caffe框架,图像数据转换成LMDB数据格式

小码农最近在研究深度学习，对所学知识做点记录，以供以后翻阅。在Caffe框架中，数据的格式都是LMDB的，如何将图像数据转换成这个格式呢？

首先，将图像数据和标签生成txt文档，执行一下代码：

find `pwd`/examples/images -type f -exec echo {} \; > examples/_temp/temp.txt

`pwd`/examples/images 是图像数据的路径，<pre name="code" class="cpp">examples/_temp/temp.txt 是保存图像数据和标签的文档，生成之后需添加标签，每行空一个填标签

然后，使用script GLOG_logtostderr=1 将图像数据转换成LMDB格式，代码如下：

EXAMPLE=examples/xx
DATA=data/xx
TOOLS=build/tools

TRAIN_DATA_ROOT=/home/lei/project/caffe/caffe-master/data/xx/train/

echo "Creating train lmdb..."

GLOG_logtostderr=1 $TOOLS/convert_imageset \
    $TRAIN_DATA_ROOT \
    $DATA/train.txt \
    $EXAMPLE/ilsvrc12_train_lmdb
echo "Done!"

$TOOLS/convert_imageset \ 转换的执行文件
<pre name="code" class="cpp">$TRAIN_DATA_ROOT \ 图像数据路径
<pre name="code" class="cpp">$DATA/train.txt \ 前面生成的文档,把temp.txt复制过来
<pre name="code" class="cpp">$EXAMPLE/ilsvrc12_train_lmdb 保存生成的LMDB数据结果

这样就生成了LMDB数据～