ImageSets文件夹中的train.txt生成路径文件train.txt

主要是让yolov5系列代码读取到我们的数据集的路径

import os
imageset = ['train','val']
labelpath = '/home/tzf/datasets/RSDD-SAR/'
for sets in imageset:
    print(sets)
    # 检查文件是否存在，存在则删除
    txt_file_address = labelpath + sets + '.txt'
    if not os.path.exists(txt_file_address):
        print('no this txt')
        # os.mkdir(txt_file_address)
    with open(labelpath +'ImageSets/Main/'+ sets + '.txt','r',encoding= 'UTF-8')as f:
        with open(txt_file_address, "w") as file:
            for i in f.readlines():
                print(i[:-1]  +'.jpg')
                imgpath = labelpath + 'JPEGImages/'+ i[:-1] +'.jpg'
                print(imgpath)
                assert os.path.exists(imgpath)
                file.write(imgpath +'\n')