Caffe训练:用shell脚本生成图片列表txt文件

针对 Caffe学习笔记(三)中 kaggle 数据集二分类问题生成的 list 代码

参考:
Shell脚本生成图片清单
shell 的cut 命令用法
Linux Shell脚本编程--sed命令详解
linux shell 用sed命令在文本的行尾或行首添加字符

创建一个 sh 脚本文件:

vi create_list.sh

输入以下代码:

#!/usr/bin/env sh    # 是一个约定的标记,它告诉系统这个脚本需要什么解释器来执行,即使用哪一种Shell
DATA=data/kaggle   # 图片保存的根目录
LIST=data/kaggle   # 生成文件的保存目录
 
echo "Create train.txt..."
rm -rf $LIST/train.txt
find $DATA/train/cat_train -name *.jpg | cut -d '/' -f4-5 | sed "/./{s/^/\//;s/$/ 0/}">>$LIST/train.txt
find $DATA/train/dog_train -name *.jpg | cut -d '/' -f4-5 | sed "/./{s/^/\//;s/$/ 1/}">>$LIST/train.txt

echo "Create val.txt..."
rm -rf $LIST/val.txt
find $DATA/val/cat_val -name *.jpg | cut -d '/' -f4-5 | sed "/./{s/^/\//;s/$/ 0/}">>$LIST/val.txt
find $DATA/val/dog_val -name *.jpg | cut -d '/' -f4-5 | sed "/./{s/^/\//;s/$/ 1/}">>$LIST/val.txt

echo "All done"

运行 shell 脚本文件:

chmod +x ./test.sh  #使脚本具有执行权限
./test.sh           #执行脚本

注意,一定要写成 ./test.sh,而不是 test.sh,运行其它二进制的程序也一样,直接写 test.sh,linux 系统会去 PATH 里寻找有没有叫 test.sh 的,而只有 /bin, /sbin, /usr/bin,/usr/sbin 等在 PATH 里,你的当前目录通常不在 PATH 里,所以写成 test.sh 是会找不到命令的,要用 ./test.sh 告诉系统说,就在当前目录找。

备注

sed 后多个命令进行连接:sed "/./{s/^/\//;s/$/ 0/}"
表示在每一行开头加上 “/”,在每一行最后加上 “ 0”,注意这里的特殊字符 /,要打印输出需要加上转义字符反斜杠 “\”

上一篇:caffe配置过程


下一篇:jupyter-notebook:在cell中运行python文件使用的python解释器和kernel不一致