使用自己的資料集“煮caffe”時候需要生成train.txt 和 val.txt。但是，沒有找到一個有效的方法。今天跟大家分享一下使用shell自動生成train.txt 和 val.txt

指令碼沒有容錯性 謹慎使用 勿噴
無原理，看步驟。

1 資料集格式要求

檔案佈局
這樣佈局也是caffe的要求，shell 下邊會貼出來或者→shell指令碼下載地址

train中

train資料夾下的圖片

val中結構
說明： val需要有train一致的結構外，還需要把各個檔案中的圖片，複製到val檔案下。

2 Shell 程式碼

filepath2txt.sh

#!/usr/bin/env sh
# 說明 
# dictionary： train{d1{f1,f2,...},d2{f1,f2,...},...} val{d1{f11,f12,...},d2{f21,f22,...},.... f11,f12,.....,f21,f2,...,....}
# train
#    -d1
#       f1.jpeg
#       f2.jpeg
#       f3.jpeg
#       f4.jpeg
#       ....
#    -d2
#       f1.jpeg
#       f2.jpeg
#       f3.jpeg
 

#       f4.jpeg
#       ....
#    ...
# val
#    -d1
#    -d2
#    ...
#    all images in di need copy to val/ again.
rm -f train.txt
rm -f val.txt
echo "filepath2traintxt"
sh ./filepath2traintxt.sh
echo "Done"
echo "filepath2valtxt"
sh ./filepath2valtxt2.sh
echo "Done"

filepath2traintxt.sh

#!/usr/bin/env sh
 

# 深度遍歷
deepls(){
    for x in "$1/"*
    do 
        #echo $x
        if [ -f $x ]
        then
            echo $x $I|cut -d '/' -f3-4 >> $NAME
        fi
        if [ -d $x ]
        then
            (deepls "$x")
            I=`expr $I + 1`
        fi
    done
}
I=0
DEST_PATH="./train"
NAME="./train.txt"
deepls $DEST_PATH

filepath2valtxt2.sh

#!/usr/bin/env sh
# 深度遍歷
deepls(){
    for x in "$1/"*
    do 
        #echo $x
        if [ -f $x ]
        then
            echo $x $I|cut -d '/' -f4-5 >> $NAME
        fi
        # if [ -d $x] error
        done
}
deep(){
    for x in "$1/"*
    do
        if [ -d $x ]
        then
            deepls "$x"
            I=`expr $I + 1`
        fi
    done
}
I=0
DEST_PATH="./val"
NAME="./val.txt"
deep $DEST_PATH

3 結果展示