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

 参考  PASCAL VOC格式的目标检测数据集生成ImageSet/Main中的各类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')