本文主要是介绍从B导的yolox、yolov7-tiny的标签中提取出来ground truth，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！
原始的test.txt内容如下：
提取出来的内容如下：
import osif __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() # 也可以直接替换成类别listoutpath = '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)
这篇关于从B导的yolox、yolov7-tiny的标签中提取出来ground truth的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
