COCO类别标签增加80

最新推荐文章于 2025-06-26 11:38:37 发布

计算机CV民工

最新推荐文章于 2025-06-26 11:38:37 发布

阅读量289

点赞数

CC 4.0 BY-SA版权

分类专栏：小Tips 文章标签： YOLO

本文链接：https://blog.youkuaiyun.com/qq_41701723/article/details/134603704

小Tips 专栏收录该内容

60 篇文章

订阅专栏

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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] + " 修改完成")

在这里插入图片描述