YOLO数据集label标号修改脚本

最新推荐文章于 2025-01-04 12:02:36 发布

Eagle104fred

最新推荐文章于 2025-01-04 12:02:36 发布

阅读量1.4k

点赞数 1

分类专栏： python 文章标签： python

本文链接：https://blog.csdn.net/qq_26625581/article/details/114163101

版权

python 专栏收录该内容

13 篇文章

订阅专栏

博客介绍了使用Python修改COCO标签标号的方法。通过运行python COCORelabel.py [要修改的数字标号] [修改为什么数字] 命令即可实现，如python COCORelabel.py 0 1 可将所有class 0 修改为class 1。

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

修改前面的label标号：
在这里插入图片描述

GetStarted

python COCORelabel.py [要修改的数字标号] [修改为什么数字]
e.g. python COCORelabel.py 0 1 (把所有的class 0 修改class 1)

import os
import sys

originNum = int(sys.argv[1])#第一个参数输入起始文件序号
reNumber = int(sys.argv[2]) #第二个参数输入修改的文件目录

txtfilepath = "./labels" #原始txt文件所存文件夹，文件夹可以有一个或多个txt文件
savefilepath = "./labels" #更改后txt文件存放的文件夹

originFileList = os.listdir(txtfilepath) # 返回指定的文件夹包含的文件或文件夹的名字的列表

num = len(originFileList)
list = range(num) #创建从0到num的整数列表 
files = os.listdir(savefilepath)
for i in list: #遍历每一个文件    
    name = originFileList[i]#拿出文件名
    readfile = open(txtfilepath+"/"+name, 'r') #读取文件
    fline = readfile.readlines() #读取txt文件中每一行
    savetxt = open(savefilepath+"/"+name,'w')
    textList=[]
    for j in fline:#每一行循环
        textList=j.split(" ",1)#分割第一个字符和后面的字符
        firstStr=textList[0]
        if firstStr.isdigit and int(firstStr)==originNum:
            retext=str(reNumber)+' '+textList[1]#rew为修改好的字符行
            print(name+":"+retext)
            savetxt.write(retext)