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')