COCO类别标签增加80

最新推荐文章于 2024-12-28 11:21:20 发布

原创最新推荐文章于 2024-12-28 11:21:20 发布 · 331 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#YOLO

小Tips 专栏收录该内容

60 篇文章

订阅专栏

本文介绍了一个Python脚本，用于处理COCO数据集中的类别标签，将原有的标签值增加80。脚本遍历train_pbr目录下的txt文件，逐行读取并更新标签，最后保存修改后的文件。

部署运行你感兴趣的模型镜像

COCO类别标签增加80

import codecs
import os

path = 'H:/Dataset/COCO/train_pbr/000001/labels/'  # 标签文件train路径
m = os.listdir(path)
# 读取路径下的txt文件
for n in range(0, len(m)):
    t = codecs.open('H:/Dataset/COCO/train_pbr/000001/labels/' + m[n], mode='r', encoding='utf-8')
    line = t.readline()  # 以行的形式进行读取文件
    list1 = []
    while line:
        a = line.split()
        list1.append(a)
        line = t.readline()
    t.close()

    lt = open('H:/Dataset/COCO/train_pbr/000001/labelNew/' + m[n], "w")
    for num in range(0, len(list1)):
        list1[num][0] = str(int(list1[num][0])+79)  # 第一列为0时，将0改为1
        lt.writelines(' '.join(list1[num]) + '\n')  # 每个元素以空格间隔，一行元素写完并换行
    lt.close()
    print(m[n] + " 修改完成")