修改txt初始数值

import os
import re

# 路径
path = 'C:\\Users\\1\\Desktop\\test\\'
# 文件列表
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('^1', '0', f.read(), flags=re.MULTILINE)   #将列中的1替换为0
    with open(file, 'w') as f:
        f.write(new_data)

yolo识别的标签是txt格式，通过修改第一个数值可以达到修改标签的效果。