问题一:
Yolov5 打标时,如果是几个人分工打标,那么当都打的是单目标的,只有一个类别的。那么.txt标签文件中的序号都是0.此时不能将他们直接合并在一起作为数据集训练。而是需要先将txt中的序号进行批量修改。然后再进行合并。查看无误后再进行训练。
下面是批量修改代码:
import os
import re
# 路径
path = 'F:/AAAYOLO/xiugai_txt/3/'
# 文件列表
files = []
for file in os.listdir(path):
if file.endswith(".txt"):
files.append(path+file)
# 逐文件读取-修改-重写
for file in files:
with open(file, 'r') as f:
new_data = re.sub('^2', '3', f.read(), flags=re.MULTILINE) # re.sub('^序号A', '序号B', f.read(), flags=re.MULTILINE) 将序号批量改为序号B
with open(file, 'w') as f:
f.write(new_data)