本文共 979 字,大约阅读时间需要 3 分钟。
修改前面的label标号:
python COCORelabel.py [要修改的数字标号] [修改为什么数字]
e.g. python COCORelabel.py 0 1 (把所有的class 0 修改class 1)
import osimport sysoriginNum = 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)
转载地址:http://fbbnn.baihongyu.com/