基于caffe的图片数据创建LMDB文件的脚本

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

#EXAMPLE和DATA设置为项目路径，TOOLS为caffe的tools路径，因为制作中需要用到caffe的工具
EXAMPLE=/home/admin01/51CTO/train-face
DATA=/home/admin01/51CTO/train-face
TOOLS=/home/matt/Documents/caffe/build/tools

#训练集和测试集的路径
TRAIN_DATA_ROOT=/home/admin01/51CTO/train-face/train/
VAL_DATA_ROOT=/home/admin01/51CTO/train-face/val/

# Set RESIZE=true to resize the images to 60 x 60. Leave as false if images have
# already been resized using another tool.
#是否要对图片进行resize以达到统一大小
RESIZE=true
if $RESIZE; then
  RESIZE_HEIGHT=227
  RESIZE_WIDTH=227
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_face_48.sh to the path" \
       "where the face_48 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_face_48.sh to the path" \
       "where the face_48 validation data is stored."
  exit 1
fi

echo "Creating train lmdb..."

#修改DATA为存放 <文件名,Label>的文本文件(不一定要是txt)
#LMDB适合单label数据
#修改EXAMPLE为要生成的LMDB的文件夹名称，文件夹内是data.mdb和lock.mdb
GLOG_logtostderr=1 $TOOLS/convert_imageset \
    --resize_height=$RESIZE_HEIGHT \
    --resize_width=$RESIZE_WIDTH \
    --shuffle \
    $TRAIN_DATA_ROOT \
    $DATA/train.txt \
    $EXAMPLE/face_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/face_val_lmdb

echo "Done."
Status API Training Shop Blog About