利用python和shell腳本生成train.txt的標簽文件

1. 用shell腳本生成帶絕對路徑的train.txt

例如我要生成如下形式的帶標簽的文件，如圖:(如有兩個標簽：0 和 1)

shell腳本如下：

這樣標簽0寫入了train.txt

# /usr/bin/env sh
DATA=/home/ttwang/Desktop/Texture/0
DATASAVE=/home/ttwang/Desktop  4 
echo "Create train.txt..."

# /usr/bin/env sh
find $DATA -name Grass*.jpg | sed "s/$/ 1/" >>$DATASAVE/train.txt  9 
echo "Done.."

再寫一個腳本，這樣就把指定文件下的想要做成標簽1的圖片寫入了train.txt，（這樣比較麻煩，肯定有比這更好的）

 

# /usr/bin/env sh
DATA=/home/ttwang/Desktop/Texture/1 DATASAVE=/home/ttwang/Desktop echo "Create train.txt..."

# /usr/bin/env sh
find $DATA -name Road*.jpg | sed "s/$/ 1/" >>$DATASAVE/train.txt echo "Done.."

 

 

然后，運行腳本： sh 1.sh / sh 2.sh (假設腳本文件名為這個)，就行了！

(1)如果不想顯示那么多路徑，就相應該DATA路徑的表示，或者使用cut 具體參考（用到時再細看）

用python腳本生成的train.txt

2.用python腳本生成train.txt

例如我要生成如下形式的帶標簽的文件，如圖:

python腳本如下：

import os
def generate(dir,label):
    files = os.listdir(dir)
    files.sort()
    print '****************'
    print 'input :',dir
    print 'start...'
    listText = open(dir+'/'+'list.txt','w')
    for file in files:
        fileType = os.path.split(file)
        if fileType[1] == '.txt':
            continue
        name = file + ' ' + str(int(label)) +'\n'
        listText.write(name)
    listText.close()
    print 'down!'
    print '****************'

if __name__ == '__main__':
    generate('/home/ttwang/Desktop/Texture/Test/1',1)