数据集打标签
import os
def generate(dir, label):
files = os.listdir(dir)
files.sort()
print('****************')
print('input :', dir)
print('start...')
listText = open(r'C:\Users\Think\Desktop\webcam.txt', 'a+')
folder1 = 'webcam/images'
for file in files:
fileType = os.path.split(file)
if fileType[1] == '.txt':
continue
name = folder1 + '/' + folder+ '/' +file + ' ' + str(int(label)) + '\n'
listText.write(name)
listText.close()
print('down!')
print('****************')
outer_path = r'C:\Users\Think\Desktop\webcam\images'
if __name__ == '__main__':
i = 0
folderlist = os.listdir(outer_path)
for folder in folderlist:
generate(os.path.join(outer_path, folder), i)
i += 1