批量修改YOLO格式的标注类别

1.解决的问题

假如你有一个YOLO格式的数据集，标注类别为0，1，2，3四个类别标签。如果你想删除标签1，只保留0，2，3类别的标注信息，或者想将标签0和标签1合并为标签1，只剩下标签1，2，3。等类似的标签修改问题

2.python代码实现

import os

# 原始txt文件存放的路径
txt_folder = "/dataset/source-sj/train/labels"
# 修改后txt文件的保存路径
new_txt_folder = "/dataset/source-sj/train/labels-1"

# 确保新的保存文件夹存在，不存在则创建
if not os.path.exists(new_txt_folder):
    os.makedirs(new_txt_folder)