os.walk不在子目录中保存图像

import os
import shutil

def save_images_from_directory(directory):
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith(".jpg") or file.endswith(".png"):
                # 处理图像文件的逻辑，例如保存到指定目录
                image_path = os.path.join(root, file)
                destination_path = "path/to/save/images"
                shutil.copy(image_path, destination_path)

# 调用函数并传入目录路径
save_images_from_directory("path/to/directory")