文件夹图片提取

1、提取当前目录下所有文件夹内的图片。（不包括二级子文件夹）

import os
import shutil

# 设置源文件夹路径和目标文件夹路径
current_dir = os.getcwd()  # 获取当前目录
target_dir = os.path.join(current_dir, 'all_images')

# 创建目标文件夹（如果不存在）
if not os.path.exists(target_dir):
    os.makedirs(target_dir)

# 图片文件扩展名列表
image_extensions = ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff']

# 遍历所有文件夹并复制图片文件
for folder_name in os.listdir(current_dir):
    folder_path = os.path.join(current_dir, folder_name)
    if os.path.isdir(folder_path):  # 检查是否为文件夹
        for filename in os.listdir(folder_path):
            if any(filename.lower().endswith(ext) for ext in image_extensions):
                source_file_path = os.path.join(folder_path, filename)
                target_file_path = os.path.join(target_dir, filename)
                shutil.copy2(source_file_path, target_file_path)

print("所有图片文件已复制到目标文件夹")

2、提取当前目录下的图片以及所有子文件夹内的图片

import os
import shutil

# 设置源文件夹路径和目标文件夹路径
current_dir = os.getcwd()  # 获取当前目录
target_dir = os.path.join(current_dir, 'all_images')

# 创建目标文件夹（如果不存在）
if not os.path.exists(target_dir):
    os.makedirs(target_dir)

# 图片文件扩展名列表
image_extensions = ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff']

def copy_images_from_folder(folder_path):
    # 遍历当前文件夹
    for item in os.listdir(folder_path):
        item_path = os.path.join(folder_path, item)
        if os.path.isdir(item_path):
            # 如果是文件夹，则递归调用
            copy_images_from_folder(item_path)
        elif any(item.lower().endswith(ext) for ext in image_extensions):
            # 如果是图片文件，则复制到目标文件夹
            target_file_path = os.path.join(target_dir, item)
            shutil.copy2(item_path, target_file_path)

# 从当前目录开始
copy_images_from_folder(current_dir)

print("所有图片文件已复制到目标文件夹")