调整txt文件中缺陷对应位置坐标，分批标注各classes.txt标签顺序不同调整至顺序一致

最新推荐文章于 2024-03-18 21:57:20 发布

ye1256

最新推荐文章于 2024-03-18 21:57:20 发布

阅读量333

点赞数 1

分类专栏：深度学习文章标签： 1024程序员节

本文链接：https://blog.youkuaiyun.com/ye1256/article/details/127499833

版权

import os
import re

outputpath = r'C:\Users\yewenjing\Desktop\1024'  # 转换后label位置
labelpath = r'C:\Users\yewenjing\Desktop\1022新标注\20221021_2\labels_172835_1027022'  # 需要转换的label位置

# source_label0 = '0'
# new_label0 = '1'

source_label1 = '1'
new_label1 = '4'

source_label2 = '2'
new_label2= '3'
#
source_label3 = '3'
new_label3= '2'

source_label4 = '4'
new_label4 = '1'
# 读取labelpath下的标签
labellist = os.listdir(labelpath)
for i in labellist:
    with open(os.path.join(labelpath, i.split('.')[0] + '.txt'), 'r') as f1:
        while 1:
            lines = f1.readline()
            if not lines:
                break
            rectangle = re.findall(r"\d+\.?\d*", lines)
            # if rectangle[0] == source_label0:
            #     rectangle[0]