用Python批量替换更改掉txt文本的内容

在我们做深度学习时,对于数据集打标后可能标签的label的id需要更改,这时如果再一个一个去改或者用labelimg去改就很麻烦了,所以下面提出一种方法即用python解决。

修改前:

用Python批量替换更改掉txt文本的内容_第1张图片

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 文本地址,它的思路是把内容去切分然后进行一个第一列的内容替换。

修改后:

用Python批量替换更改掉txt文本的内容_第2张图片

你可能感兴趣的:(小技巧,python,深度学习,数据挖掘)