PYTHON:由标签文件生成训练集，测试集图像列表以及量化图像列表

import os
import random

def trainTestGen(mainDir,val_percent):
    image_labels = [f for f in os.listdir(os.path.join(mainDir,'labels'))]  # 存放labels的路径
    train_file = open(mainDir + '/train.txt', 'w')
    val_file = open(mainDir + '/val.txt', 'w')
    quaImageList=open(mainDir+'/quaImageList.txt','w')
    random.shuffle(image_labels)
    for i, image_id in enumerate(image_labels):
        if image_id[-3:] == 'txt': 
            if i < (len(image_labels) * val_percent):
                val_file.write(mainDir + '/person_pic' + '/%s\n' % (image_id[:-3] + 'jpg'))
            else:
                train_file.write(mainDir + '/person_pic' + '/%s\n' % (image_id[:-3] + 'jpg'))
                quaImageList.write(mainDir+'/rknn416'+'/%s\n'%(image_id[:-3]+'jpg'))
    train_file.close()
    val_file.close()
    quaImageList.close()

    print('[Warning] Successfully Conversion!')

if __name__ == '__main__':
    mainDir='/home/.../extra/personDataset'
    trainTestGen(mainDir,0.1)