Dataset创建之前需的txt存放Python实现

import os
a=0
while(a<5):  # a表示类别
    dir = 'img/'+str(a)+'/'
    label = a

    files = os.listdir(dir)
    files.sort() # 对文件中的数据进行排序
    train = open('train.txt','a')
    val = open('test.txt', 'a')
    i = 1
    for file in files:
        if i<20: # 每个文件夹下存放小于20张图片的地址会存放在train.txt中，超过20张的将会存放在test.txt文本
            fileType = os.path.split(file)
            if fileType[1] == '.txt':
                continue
            name =  str(dir) +  file + ' ' + str(int(label)) +'\n'
            train.write(name)
            i = i+1
            print(i)
        else:
            fileType = os.path.split(file)
            if fileType[1] == '.txt':
                continue
            name = str(dir) +file + ' ' + str(int(label)) +'\n'
            test.write(name)
            i = i+1

    val.close()
    train.close()
    print(a)
    a = a + 1

地址

https://public.roboflow.com/object-detection/