py脚本给txt文件的每一行添加指定路径——yolo数据集格式

使用python脚本给txt文件的每一行添加指定路径

---

代码

import os

def updateFile(file, str):
    """
    在文件的每一行前面增加关键字
    :param file:文件名
    :param str:关键字
    """
    file_data = ""
    with open(file, "r") as f:
        for line in f:
            line = str + line.strip()+ ".jpg"+"\n"
            file_data += line
    with open(file, "w") as f:
        f.write(file_data)

base_dir ='/home/xxxxx/txt/val'
str = "/home/xxxxx/images/"
for file_name in os.listdir(base_dir):
    file_name = os.path.join(base_dir, file_name)
    print(file_name)
    updateFile(file_name, str)

通过给指定的txt文件添加路径以及图片格式制作yolo数据集的训练格式
其中：
base_dir为txt文件所在的目录
str为要在txt中所添加的路经信息