从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)
更多推荐
已为社区贡献21条内容
所有评论(0)