帮我写一个python脚本，要求将/root/YOLOv8/crack_datasets/images/train路径下的所有.jpg文件以路径/文件名.jpg的格式保存在/root/YOLOv8/crack_datasets下的train.txt文档中

当然可以，这是一个简单的Python脚本，它会遍历指定目录并生成包含文件路径和文件名的文本文件：

import os

# 源图片目录
source_dir = "/root/YOLOv8/crack_datasets/images/train"
# 目标txt文件路径
output_file = "/root/YOLOv8/crack_datasets/train.txt"

with open(output_file, 'w') as f:
    for root, dirs, files in os.walk(source_dir):
        for file in files:
            if file.endswith('.jpg'):
                # 构造文件路径
                file_path = os.path.join(root, file)
                # 写入txt文件
                f.write(file_path + '\n')

print("所有.jpg文件已成功写入到train.txt")

#