caffe生成LMDB时的坑

编辑此文件，写入如下代码，并保存

#!/usr/bin/env sh DATA=data/re/ MY=examples/myfile  echo "Create train.txt..." rm -rf $MY/train.txt for i in 3 4 5 6 7  do find $DATA/train -name $i*.jpg | cut -d '/' -f4-5 | sed "s/$/ $i/">>$MY/train.txt done echo "Create test.txt..." rm -rf $MY/test.txt for i in 3 4 5 6 7 do find $DATA/test -name $i*.jpg | cut -d '/' -f4-5 | sed "s/$/ $i/">>$MY/test.txt done echo "All done"

然后，运行此脚本

# sudo sh examples/myfile/create_filelist.sh

成功的话，就会在examples/myfile/ 文件夹下生成train.txt和test.txt两个文本文件，里面就是图片的列表清单。一定要生成这种格式，把train带上

不然下面生成lmdb会报路径错误

情况二

#!/usr/bin/env sh
DATA=data/mnist/
MY=examples/myfile

echo "Create train.txt..."
rm -rf $MY/train.txt
for i in 0 1 2 3 4 5 6 7 8 9
do
find $DATA/train/$i/ -name *.png | cut -d '/' -f4-6 | sed "s/$/ $i/">>$MY/train.txt
done

echo "Create test.txt..."
rm -rf $MY/test.txt
for i in 0 1 2 3 4 5 6 7 8 9
do
find $DATA/test/$i/ -name *.png | cut -d '/' -f4-6 | sed "s/$/ $i/">>$MY/test.txt
done
echo "All done"

执行命令 一定要在data之前的文件夹执行，sudo sh 到createfile.sh

一定要生成这种格式不然错误同上。

转载请标明出处:caffe生成LMDB时的坑

文章来源: caffe生成LMDB时的坑