为机器人换个好使的脑子(十五)

接下来,我们要生成lmdb数据。在caffe/data下创建Sean目录,及create_data.sh

$ ~/caffe/data/Sean/create_data.sh
# create_data.sh
root_dir="$HOME/caffe"
cd $root_dir
redo=1
dataset_name="Sean"
data_root_dir="$HOME/$dataset_name"
mapfile="$data_root_dir/labelmap.prototxt"
anno_type="detection"
db="lmdb"
min_dim=0
max_dim=0
width=0
height=0

extra_cmd="--encode-type=jpg --encoded"
if [ $redo ]
then
  extra_cmd="$extra_cmd --redo"
fi
for subset in test trainval
do
  python $root_dir/scripts/create_annoset.py --anno-type=$anno_type --label-map-file=$mapfile --min-dim=$min_dim --max-dim=$max_dim --resize-width=$width --resize-height=$height --check-label $extra_cmd $data_root_dir $data_root_dir/structure/$subset.txt $data_root_dir/$dataset_name/$db/$dataset_name"_"$subset"_"$db examples/$dataset_name
done

最终数据集

Sean/
  |- images/            图片
  |- labels/            标注
  |- structure/         图片和标注的对应文件
  |- Sean/
       |- lmdb/
            |- Sean_test_lmdb/
            |- Sean_train_lmdb/