如何在目录中搜索图像并将其保存到另一个文件夹？

import os
import shutil

def search_and_save_images(source_dir, target_dir):
    # 遍历目录
    for root, dirs, files in os.walk(source_dir):
        for file in files:
            file_path = os.path.join(root, file)
            
            # 判断文件类型
            if is_image_file(file_path):
                # 复制或移动文件
                target_path = os.path.join(target_dir, file)
                shutil.copy(file_path, target_path)
                # 或者使用以下代码移动文件
                # shutil.move(file_path, target_path)

def is_image_file(file_path):
    # 判断文件扩展名是否为图像文件
    image_extensions = ['.jpg', '.jpeg', '.png', '.gif']
    ext = os.path.splitext(file_path)[1].lower()
    return ext in image_extensions

# 示例用法
source_directory = '/path/to/source/directory'
target_directory = '/path/to/target/directory'
search_and_save_images(source_directory, target_directory)