【Caffe】shell指令将图片信息转txt训练信息文件并打乱

在caffe中训练模型需要将训练的图片打包成txt文件，这里通过linux自带的shell脚本进行转换，具体代码如下：

#!/bin/bash
# get all filename in specified path

path=$1
files=$(ls $path|head -n 80)
for filename in $files
do
 #mv $filename $path2
 echo `realpath $filename` >> train.txt
done

改进版：通过find指令对for循环进行改进：

find `pwd`/testset/NG/ -name "*.jpg" >> train.txt

 

在输出成完成之后，需要在每一行的末尾输入标签（如果是2分类输出0/1），最后通过shuf指令将内部的文件打乱：

shuf train.txt -o train.txt