小码农最近在研究深度学习,对所学知识做点记录,以供以后翻阅。在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数据~