python：yolov算法修改数据集txt标签类别

比如1换成2

import os

def modify_txt(oldtxt_path):
    for oldtxt in os.listdir(oldtxt_path):
        b = []
        txt_root=os.path.join(oldtxt_path+oldtxt)
        with open(txt_root,'r',encoding='utf-8') as f:
            _txt=f.readlines()# 读取所有行，readlines返回的是列表
            for i in _txt:
                b.append(i.split(' '))  #以空格分开
                for x in b:
                    if x[0]=='1':  #这里换成你想要改的类别
                        x[0]='2'
                #print(b)
            with open(txt_root,'w+',encoding='utf-8') as f: #把列表b里的信息再重新写入
                for c in b:
                    f.writelines(' '.join(c))
if __name__ == "__main__":
    p='D:/Postgraduate study program(cxl)/daixingguang/data module/221030-1/label-yolo/'#换成自己的路径，路径中不要出现中文，否则无法识别
    modify_txt(p)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20