Caffe下将图片转换为lmdb格式

时间:2022-11-11 21:06:30

在Caffe的例程里,自带了转换格式的代码。查看文件:./example/imagenet/create_imagenet.sh
文件内容如下:

#!/usr/bin/env sh
# Create the imagenet lmdb inputs
# N.B. set the path to the imagenet train + val data dirs
set -e

EXAMPLE=examples/imagenet
DATA=data/ilsvrc12
TOOLS=build/tools

TRAIN_DATA_ROOT=/path/to/imagenet/train/
VAL_DATA_ROOT=/path/to/imagenet/val/

# Set RESIZE=true to resize the images to 256x256. Leave as false if images have
# already been resized using another tool.
RESIZE=false
if $RESIZE; then
RESIZE_HEIGHT=256
RESIZE_WIDTH=256
else
RESIZE_HEIGHT=0
RESIZE_WIDTH=0
fi


if [ ! -d "$TRAIN_DATA_ROOT" ]; then
echo "Error: TRAIN_DATA_ROOT is not a path to a directory: $TRAIN_DATA_ROOT"
echo "Set the TRAIN_DATA_ROOT variable in create_imagenet.sh to the path" \
"where the ImageNet training data is stored."
exit 1
fi

if [ ! -d "$VAL_DATA_ROOT" ]; then
echo "Error: VAL_DATA_ROOT is not a path to a directory: $VAL_DATA_ROOT"
echo "Set the VAL_DATA_ROOT variable in create_imagenet.sh to the path" \
"where the ImageNet validation data is stored."
exit 1
fi

echo "Creating train lmdb..."

GLOG_logtostderr=1 $TOOLS/convert_imageset \
--resize_height=$RESIZE_HEIGHT \
--resize_width=$RESIZE_WIDTH \
--shuffle \
$TRAIN_DATA_ROOT \
$DATA/train.txt \
$EXAMPLE/ilsvrc12_train_lmdb

echo "Creating val lmdb..."

GLOG_logtostderr=1 $TOOLS/convert_imageset \
--resize_height=$RESIZE_HEIGHT \
--resize_width=$RESIZE_WIDTH \
--shuffle \
$VAL_DATA_ROOT \
$DATA/val.txt \
$EXAMPLE/ilsvrc12_val_lmdb

echo "Done."

其实我们只需要更改这个文件,然后把我们的文件路径对应填写进去就可以了。

最关键的语句如下:

$TOOLS/convert_imageset \
--resize_height=$RESIZE_HEIGHT \
--resize_width=$RESIZE_WIDTH \
--shuffle \
$TRAIN_DATA_ROOT \
$DATA/train.txt \
$EXAMPLE/ilsvrc12_train_lmdb

使用了/build/tools/convert_imageset工具,后面的几个参数分别是设置转换图像大小、是否随机分配图片、图像存放位置、图像对应的文件名列表、以及生成lmdb文件的存放路径。

关于TRAIN_DATA_ROOT:
图像数据可以存放在任意路径下,但是要相应地更改TRAIN_DATA_ROOT
的值。
关于train.txt:
train.txt需要生成,生成方式为:每一行表示一个文件,每行的第一项是文件名,第二项是序号。比如说,在训练数据集中有两张图片,那么train.txt的内容应该如下:

image1.jpg 1
image2.jpg 2

参考资料:
Caffe官网:Brewing ImageNet
Caffe学习系列(11)