caffe:create_txt.sh（数据预处理成txt文本文件）

最新推荐文章于 2021-12-11 18:19:17 发布

转载最新推荐文章于 2021-12-11 18:19:17 发布 · 301 阅读

caffe 专栏收录该内容

19 篇文章

订阅专栏

本文介绍了一种使用Caffe深度学习框架进行图像分类任务时的数据准备方法。通过shell脚本自动化创建训练集（train.txt）和验证集（val.txt），并对不同类别的图像文件进行路径整理及类别标记。

参考网址：http://www.cnblogs.com/carle-09/p/5778965.html

1、

#!/usr/bin/env sh
DATA=/home/wp/CAFFE/caffe-master/myself/00b
MY=/home/wp/CAFFE/caffe-master/myself/00b

echo "Creating train.txt..."
rm -rf $MY/train.txt
find $DATA/train/banana -name *.jpg| cut -d/ -f9-10 | sed "s/$/ 0/">>$MY/train.txt
find $DATA/train/cata   -name *.jpg| cut -d/ -f9-10 | sed "s/$/ 1/">>$MY/train.txt
find $DATA/train/flower -name *.jpg| cut -d/ -f9-10 | sed "s/$/ 2/">>$MY/train.txt

echo "Creating val.txt..."
rm -rf $MY/val.txt
find $DATA/val/banana  -name *.jpg| cut -d '/' -f9-10 | sed "s/$/ 0/">>$MY/val.txt
find $DATA/val/cata    -name *.jpg| cut -d '/' -f9-10 | sed "s/$/ 1/">>$MY/val.txt
find $DATA/val/flower  -name *.jpg| cut -d '/' -f9-10 | sed "s/$/ 2/">>$MY/val.txt

echo "Done."

# cd CAFFE/caffe-master
# sh ./myself/00b/create_txt.sh

2、

#!/usr/bin/env sh
DATA=data/re/
MY=data/re
 
echo "Create train.txt..."
rm -rf $MY/train.txt
find $DATA/train -name 3*.jpg | cut -d '/' -f4-5 | sed "s/$/ 0/">>$MY/train.txt
find $DATA/train -name 4*.jpg | cut -d '/' -f4-5 | sed "s/$/ 1/">>$MY/train.txt
echo "Create test.txt..."
rm -rf $MY/val.txt
find $DATA/test -name 3*.jpg | cut -d '/' -f4-5 | sed "s/$/ 0/">>$MY/val.txt
find $DATA/test -name 4*.jpg | cut -d '/' -f4-5 | sed "s/$/ 1/">>$MY/val.txt
echo "All done"