problem formulation

有时候下载的数据集如下，就很烦，一个里面就一张图片

code

import os
import shutil

# 定义源目录和目标目录
source_dir = './dataset/data/Detection'
destination_dir = './dataset/data/img'

# 确保目标目录存在，如果不存在则创建
if not os.path.exists(destination_dir):
    os.makedirs(destination_dir)

# 遍历源目录及其子目录
for root, dirs, files in os.walk(source_dir):
    for file in files:
        # 检查是否为图片文件
        if file.endswith(('.jpg', '.jpeg', '.png', '.gif', '.bmp')):
            # 构建完整源文件路径和目标文件路径
            source_file_path = os.path.join(root, file)
            destination_file_path = os.path.join(destination_dir, file)

            # 复制图片到目标目录
            shutil.copy2(source_file_path, destination_file_path)

print("图片提取完成！")