voc数据集生成test.txt

空谷_临风

已于 2022-03-07 14:30:55 修改

阅读量688

点赞数

分类专栏：笔记文章标签： python

于 2022-03-06 21:07:26 首次发布

原文链接：https://cloud.tencent.com/developer/article/1654841

版权

笔记专栏收录该内容

3 篇文章

订阅专栏

这段代码主要用于处理VOC2012数据集，首先获取Annotations目录下的所有文件名，并将它们保存到列表中。接着读取trainval.txt文件的内容，将每行内容（不包含换行符）存入另一个列表。最后，创建一个新的test.txt文件，将不在训练验证集列表中的文件名写入该文件，用于构建测试集。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import os

rootann = 'VOCdevkit\VOCdevkit\VOC2012\Annotations'
roottrainval = 'VOCdevkit\VOCdevkit\VOC2012\ImageSets\Main\\trainval.txt'
roottest = 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')