由PASCAL VOC的trainval.txt生成test.txt

import os rootann = 'D:\dataset\VOCdevkit\VOCdevkit\VOC2012\Annotations' roottrainval = 'D:\dataset\VOCdevkit\VOCdevkit\VOC2012\ImageSets\Main\\trainval.txt' roottest = 'D:\dataset\VOCdevkit\VOCdevkit\VOC2012\ImageSets\Main\\test.txt' file_names = os.listdir(rootann) alls = [] for file_name in file_names: file_name = file_name[:-4] # print(file_name) alls.append(file_name) with open(roottrainval) as f: trainvals = f.readlines() trainandvals = [] for trainval in trainvals: trainandvals.append(trainval[:-1]) # print(trainandvals) with open(roottest,'w') as f: for all in alls: if all not in trainandvals: f.write(all+'\n')

 

 

 

承接Matlab、Python和C++的编程，机器学习、计算机视觉的理论实现及辅导，本科和硕士的均可，咸鱼交易，专业回答请走知乎，详谈请联系QQ号757160542，非诚勿扰。