使用caffe的时候,需要先将图片生成描述文本,然后通过 convert_imageset 根据该描述文本生成 LMDB 或者 LEVELD 格式数据,用于Training 。
当图片数据很多的,就需要通过脚本来实现。
1. 首先,把图片放入不同类的目录下。
例如:
2. 运行脚本
#!/usr/bin/env sh
rm -rf train.txt
echo "TrainData: directory for train data "
num=7 # category number
a=0
while (($a<$num))
do
ls TrainData/data$a >train$a.txt
sed -i 's/jpg/jpg '$a' /g' train$a.txt
sed -i 's/'$a'_/TrainData\/data'$a'\/'$a'_/g' train$a.txt
cat train$a.txt >>train.txt
rm -rf train$a.txt
let "a++"
done
echo "Done!!"
3. 生成的描述文本: