数据标注: 修改txt文件内的指定内容(已验证）

高高呀~

已于 2022-04-27 11:02:06 修改

阅读量325

点赞数

CC 4.0 BY-SA版权

分类专栏：计算机视觉文章标签：人工智能深度学习计算机视觉 ieee论文目标跟踪

于 2022-04-21 15:28:43 首次发布

本文链接：https://blog.youkuaiyun.com/qwazp3526cn/article/details/124323025

计算机视觉专栏收录该内容

42 篇文章

订阅专栏

该代码片段遍历指定的数据目录，对每个子目录中的.txt文件进行读取，然后将读取到的每一行内容的特定部分（如x、y、w、h）写入新的.txt文件中，每行以数据编号开头。整个过程涉及文件路径的构建、文件打开与关闭以及文件内容的处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import os

data_dir = ['xx','xxx', 'xxxx']

for data_num, data in enumerate(data_dir):
    path_dir = './' + data + '/'

    file_list = [os.path.join(path_dir, _) for _ in os.listdir(path_dir) if _.endswith(".txt")];

    output_path = './output/' + data
    print(file_list)
    for num, i in enumerate(file_list):

        txt_open_ =  open(i, 'r')
        new_txt = open(output_path + '/' + i.split("/")[-1:][0], 'w')

        while True:
            line = txt_open_.readline()
            if not line:
                break
            x = line.split(" ")[1:][0]
            y = line.split(" ")[1:][1]
            w = line.split(" ")[1:][2]
            h = line.split(" ")[1:][3]
            new_txt.write(str(data_num) + " " + x + " " + y + " " + w + " " + h + '\n')


        new_txt.close()

        txt_open_.close()