Python-修改Yolov5标注文件-classes文件里分类类别

最新推荐文章于 2023-11-22 14:59:31 发布

米线炖土豆

最新推荐文章于 2023-11-22 14:59:31 发布

阅读量1k

点赞数 2

文章标签： python

本文链接：https://blog.csdn.net/weixin_50978072/article/details/129126185

版权

'''
说明：
    比如：下载好的车的类别现有标签为（0:car  1:bus），需要改成你的项目对应类别（4:car   10:bus）
    0 -> 4  1 -> 10

'''
import os
import re

# 原txt文件目录（不要以\结尾）
input_dir = r'输入路径'
# 改标签后txt文件目录（不要以\结尾）
out_dir = r"输出路径"

if not os.path.exists(out_dir):
    os.mkdir(out_dir)

def file_name(input_dir):
    for dir_path, _, files in os.walk(input_dir):
        for i_name in files:
            i_path = os.path.join(dir_path, i_name)
            o_path = os.path.join(out_dir, i_name)
            with open(i_path, 'r+', encoding='utf-8') as f:
                for i in f:
                    print(i)
                    # 需要换几个，写几个if(15变0)
                    if re.match('15', i):
                        with open(o_path, 'a+') as g:
                            data = '0' + i[2:]
                            g.write(data)
                    # elif re.match('16', i):
                    #     with open(o_path, 'a+') as g:
                    #         data = '1' + i[2:]
                    #         g.write(data)
                    else:
                        with open(o_path, 'a+') as g:
                            g.write(i)


if __name__ == '__main__':
    file_name(input_dir)