python脚本实现一次提取多个文件下的图片

2024-01-01 09:05:24

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("图片提取完成！")

文章来源:https://blog.csdn.net/qihshe/article/details/135316313
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。