深度学习开发过程中常用数据处理脚本（持续更新中）

将指定文件夹下的图片枚举并将路径存在列表中：

def list_images(folder_path):
    # 支持的图片扩展名
    extensions = ['.jpg', '.jpeg', '.png', '.gif']
    # 遍历文件夹中的所有文件
    image_list = []
    for root, dirs, files in os.walk(folder_path):
        for file in files:
            # 获取文件扩展名
            ext = os.path.splitext(file)[1]
            # 如果是图片文件，则将其路径添加到列表中
            if ext.lower() in extensions:
                image_list.append(os.path.join(root, file))
    return image_list

目录下的文件除扩展名与其他目录下的文件出扩展名外不相同名称的移除

import os

dir_path = "/path/to/directory"  # 指定目录路径
target_dir_path = "/path/to/target/directory"  # 另一个目录路径

# 获取目标目录下的所有文件名（不包括扩展名）
target_file_names = [os.path.splitext(f)[0] for f in os.listdir(target_dir_path)]

# 遍历指定目录下的所有文件
for file_name in os.listdir(dir_path):
    # 如果文件名（不包括扩展名）与目标目录下的某个文件名相同，则删除该文件
    if os.path.splitext(file_name)[0] in target_file_names:
        os.remove(os.path.join(dir_path, file_name))