在我们做深度学习时,对于数据集打标后可能标签的label的id需要更改,这时如果再一个一个去改或者用labelimg去改就很麻烦了,所以下面提出一种方法即用python解决。
修改前:
import os, glob
if __name__ == '__main__':
txt_list = glob.glob("E:/bzl_online_dataset/2020-07-20/imgs/*.txt")
for txt_item in txt_list:
with open(txt_item) as f:
lines = f.readlines()
with open(txt_item, 'w') as f:
for line in lines:
line_split = line.strip().split()
line_split[0] = '0'
f.write(
line_split[0] + ' ' +
line_split[1] + " " +
line_split[2] + " " +
line_split[3] + " " +
line_split[4]+'\n')
pass
上面的路径就是你的 txt 文本地址,它的思路是把内容去切分然后进行一个第一列的内容替换。
修改后: