python批量将txt中的标签0变成标签1

在数据集标注中,将检测物体标注为1,标注完后觉得1不合适,又想将全部的标签文件中的目标1改成2,txt的修改方法,如下图所示 

import os

def makexml(txtPath):  # 读取txt路径,xml保存路径,数据集图片所在路径
        files = os.listdir(txtPath)
        for i, name in enumerate(files):
            txtname= txtPath + name
            #使用with open,不用close
            with open(txtname) as txtFile:
                txtList = txtFile.readlines()
            with open(txtname, 'w') as f:
              for line in txtList:
                  line_split = line.strip().split()
                  if line_split[0] == '2':
                      line_split[0] = '1'
                  f.write(
                      line_split[0] + ' ' +
                      line_split[1] + " " +
                      line_split[2] + " " +
                      line_split[3] + " " +
                      line_split[4] + '\n')
            pass


if __name__ == "__main__":
    txtPath = "F:/bishe/project/yolov5-mangren/VOCdevkit/labels2change/val/"
    makexml(txtPath)

  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值