从txt标签文件生成lmdb(其实是得到一个文件夹,里面有2个文件)
current_path=`cd $(dirname $0) ; pwd ` #获取脚本所在路径
EXAMPLE=$current_path #lmdb要放的路径
DATA=$current_path #txt文件所在路径
TOOLS=/home/cdli/caffe-ssd/build/tools
TRAIN_DATA_ROOT=/
VAL_DATA_ROOT=/
# 这里我们打开resize,需要把所有图片尺寸统一
RESIZE=true
if $RESIZE; then
RESIZE_HEIGHT=256
RESIZE_WIDTH=256
else
RESIZE_HEIGHT=0
RESIZE_WIDTH=0
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/train_lmdb #生成的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/val_lmdb #生成的lmdb路径
echo "Done."