从B导的yolox、yolov7-tiny的标签中提取出来ground truth

原始的test.txt内容如下：

提取出来的内容如下：

import os

if __name__ == '__main__':
    # 将每行的内容单独读取到一个列表中
    with open('test.txt', 'r') as f:    # 这个是Main里面的test.txt内容
        datalist = f.readlines()
        data = []
        for d in datalist:
            data.append(d.strip().split())

    labelnames = open('model_data/voc_classes.txt').read().strip().split()  # 也可以直接替换成类别list
    outpath = 'map_out_2/caogao'

    for d in data:  # 每一行
        imagename = d[0].split('\\')[-1].split('.')[0]
        # 打开这个名称对应的txt写入内容
        objpath = os.path.join(outpath, f'{
      
      imagename}.txt')
        with open(objpath, 'w') as f:
            for obj in d[1:]:  # 图像名称之后的所有数据
                objlist = obj.strip().split(',')
                f.write(labelnames[int(objlist[-1])])
                f.write(" " + " ".join([a for a in objlist[:-1]]) + '\n')

    print(data)