文件夹名称作为类别并且将对应文件夹下面的图片包括路径输出到txt文件中
感谢这位博主 http://www.cnblogs.com/denny402/p/5684431.html
mnist图片数据下载:http://pan.baidu.com/s/1pLMV4Kz
我的文件夹存放是~/mnist/train/9/*.png形式存放,数字9对应图片的标签,也就是9这个文件夹里的图片都是9这个字
下面代码就是自动找出9并在txt文件中输出
#!/usr/bin/env sh
DATA=/home/xhj/hjxu-code/matlabcode/MNIST-TEST/mnist/train
SavePath=/home/xhj/hjxu-code/matlabcode/MNIST-TEST/mnist/profile
rm -rf $SavePath/train.txt
echo "Create train.txt..."
for file_a in ${DATA}/*
do
temp_file=`basename $file_a`
path1=$DATA/${temp_file}
echo $temp_file
echo $path1
#cd $path1
find $path1 -name *.png | cut -d '/' -f7-10 | sed "s/$/ $temp_file/">>$SavePath/train.txt
# echo $temp_file
done
echo "Done.."
编辑列表后在下一章制作lmdb格式时发现
Could not open or find file /home/xhj/hjxu-code/matlabcode/MNIST-TEST/mnist/test/mnist/test/5/06600.png的错误,这是由于列表txt文件列表不对,于是
cut -d ‘/’ -f9-10改了下就可以了