（LORA训练）（图片名名复制至txt内）LORA训练打标程序

把文件夹与子文件夹下所有图片建立txt文本，文本名与文本内部的内容以图片名为准，为lora训练做准备。

---

 

import os

def create_txt_from_image():
    # 请求用户输入文件夹地址
    root_folder = input("请输入图片所在文件夹的完整路径：")

    # 判断路径是否存在
    if not os.path.exists(root_folder):
        print("路径不存在，请检查输入的地址。")
        return

    # 使用os.walk遍历文件夹及其所有子文件夹
    for folder_path, dirs, files in os.walk(root_folder):
        for file in files:
            # 检查文件是否为图片（这里我们检查几种常见的图片格式）
            if file.endswith(('.jpg', '.png', '.jpeg')):
                # 获取不带扩展名的文件名
                base_name = os.path.splitext(file)[0]

                # 创建同名的txt文件路径
                txt_path = os.path.join(folder_path, base_name + '.txt')

                # 将图片文件名（不包括后缀）写入到txt文件中
                with open(txt_path, 'w', encoding='utf-8') as txt_file:
                    txt_file.write(base_name)

    print("所有图片对应的txt文件已创建完毕。")

# 运行函数
create_txt_from_image()