原本的标签的classes.txt文件中person类别是1【即第二行才是person类】,而后来找到的数据集大且标注好了,好巧不巧person类别是0。故要将labels文件的类别都改成0。
要自己先创建好空的文件夹存放新标签。
import os
import random
import numpy as np
from numpy import *
txtfilepath = "/content/drive/MyDrive/person/labels/test"
savefilepath = "/content/drive/MyDrive/person/labels/TEST"
total_txt = os.listdir(txtfilepath)
num = len(total_txt)
list = range(num)
files = os.listdir(savefilepath)
for i in list:
name = total_txt[i]
readfile = open(txtfilepath+"/"+name, 'r')
fline = readfile.readlines()
savetxt = open(savefilepath+"/"+name,'w+')
for temp in fline:
list1=temp.split()
list1[0] = "0"
b = " ".join(list1)
savetxt.write(b)
savetxt.write('\n')
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)