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

最新推荐文章于 2023-11-23 13:41:30 发布

ye1256

最新推荐文章于 2023-11-23 13:41:30 发布

阅读量262

点赞数 1

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

本文链接：https://blog.csdn.net/ye1256/article/details/127499833

版权

深度学习专栏收录该内容

11 篇文章 2 订阅

订阅专栏

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] = new_label0
            if rectangle[0] == source_label1:
                rectangle[0] = new_label1
            elif rectangle[0] == source_label2:
                rectangle[0] = new_label2
            elif rectangle[0] == source_label3:
                rectangle[0] = new_label3
            elif rectangle[0] == source_label4:
                rectangle[0] = new_label4
            with open(os.path.join(outputpath, i.split('.')[0] + '.txt'), 'a+') as f2:
                f2.write(" ".join(i for i in rectangle) + '\n')

print('ok')