针对许多类图片训练集，测试集能够覆盖所有类别生成test，val和train

最新推荐文章于 2024-04-23 08:53:13 发布

逼疯代码

最新推荐文章于 2024-04-23 08:53:13 发布

阅读量454

点赞数 1

本文链接：https://blog.csdn.net/qq_40806289/article/details/95457698

版权

1.首先得到所有的文件名称(这里获取所有的xml文件，其中xml文件名已类别+序列命名)

cd Annotations_Root
find ./ -name "*" |sort >>../ImageSets/Main/all.txt

进入标签xml的文件夹，将所有文件保存在all.txt中。

import os

num = 0
for tmpline in open("./all.txt"):
    num += 1
#    print num
    if num==1:
        continue
    tmp = tmpline.strip('\n')
    tmp = tmp.replace(".xml","")
    tmp = tmp.replace("./","")
    tmpp=tmp.split('_',3)
    number= int(tmpp[2])
    aa = number%10
    if aa==0 or aa==1:
	with open('/data_2/data/train_data/ss_big_obj_object_bdf/ssd_big_obj/ImageSets/Main/test.txt', 'a') as f:
	    f.writelines(tmp + '\n')
    elif aa==2:
	with open('/data_2/data/train_data/ss_big_obj_object_bdf/ssd_big_obj/ImageSets/Main/val.txt', 'a') as f:
	    f.writelines(tmp + '\n')
    else:
	with open('/data_2/data/train_data/ss_big_obj_object_bdf/ssd_big_obj/ImageSets/Main/trainval.txt', 'a') as f:
	    f.writelines(tmp + '\n')

将序号尾号为0或者1的文件作为测试集;

将序号尾号为2的文件作为验证集;

将序号尾号为其它(除了0,1,2)的作为训练集。