做语义分割任务,个人原本的数据集是VOC格式的,后来转换成YOLO的txt格式,原先数据集是划分好的,训练集,测试集对应的图片序号记录在一个txt文件里,需要从txt中读取相应的数据,然后复制总标签文件加下的相对应的图片标签到另外一个文件夹下。
自己写的代码如下,记录和分享一下。
from shutil import copy
im_num = [] # 创建一个列表
for line in open("E:/Dataset/VOCdevkit/VOC2007/ImageSets/Main/train.txt", "r"): # 以只读方式打开txt文件。文件的指针将会放在文件的开头。
im_num.append(line) # 将读取的数字导入im_num列表中
for a in im_num: # 遍历im_num列表
txt_name = 'F:/coco/labels/train2017/{}'.format(a[:-1]) + '.txt' # 原txt的保存路径
print(txt_name)
tar_name = 'F:/coco/labels/train2018/{}'.format(a[:-1]) + '.txt' # 选择出来后的txt的保存路径
copy(txt_name,tar_name)