Yolov5的随手记

问题一：
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)