实用深度学习图像前处理脚本分享4--xml和jpg文件分开

有时候下载到的数据xml和jpg是混合在一起的，这时候就需要将其分开

import os
import shutil

# 指定包含jpg和xml文件的文件夹路径
folder_path = r'your_path'

# 创建存放jpg文件的文件夹
jpg_folder = os.path.join(folder_path, 'jpg_files')
os.makedirs(jpg_folder, exist_ok=True)

# 创建存放xml文件的文件夹
xml_folder = os.path.join(folder_path, 'xml_files')
os.makedirs(xml_folder, exist_ok=True)

# 遍历文件夹中的文件
for file_name in os.listdir(folder_path):
    if file_name.endswith('.jpg'):
        shutil.move(os.path.join(folder_path, file_name), os.path.join(jpg_folder, file_name))
    elif file_name.endswith('.xml'):
        shutil.move(os.path.join(folder_path, file_name), os.path.join(xml_folder, file_name))

print('jpg和xml文件已成功分开。')