批量修改yolo数据集格式标签

import os

olddir = "~/labels/val/"  #原标签目录
newdir = "~/labels/val_new/"  #新标签目录

if not os.path.exists(newdir):
    os.makedirs(newdir)

all_oldtxt = os.listdir(olddir)
for oldfile in all_oldtxt:
    #read
    with open(olddir + oldfile,"r") as f:
        lines=f.readlines() 
        data=[]  
        for i in lines:
            i_list = i.split(" ")
            ## --------------修改标签---------------
            # if i_list[0] == "0" or i_list[0] == "1": #序号根据实际情况修改
            #     i_list[0] = "1"
            #     i_str = " ".join(i_list)
            # elif i_list[0] == "3":
            #     i_list[0] = "0"
            #     i_str = " ".join(i_list)
            # else:
            #     continue
            ## -------------删除标签------------------
            if i_list[0] == "5":       #序号根据实际情况修改
                print(oldfile)
                continue
            elif i_list[0] == "4":
                print(oldfile)
                continue
            else:
                i_str = " ".join(i_list)
            data.append(i_str)     #记录每一行      
    #write
    newfile = newdir + oldfile
    with open(newfile,"w") as f:
        for i in data:
            f.writelines(i)