多文件夹随机抽取图片并另存多个文件夹

ALSJL

已于 2022-08-05 10:54:59 修改

阅读量1.3k

点赞数

文章标签： python

于 2022-02-03 14:37:47 首次发布

本文链接：https://blog.csdn.net/liiojhggg/article/details/122775801

版权

这段Python代码用于遍历指定目录下的子文件夹，创建新的目标文件夹，并随机选取每个子文件夹内一定数量的图片进行复制。它使用os和shutil库来操作文件和目录，随机选取图片则通过random模块实现。关键步骤包括获取文件夹列表、创建目标路径、随机选取样本图片以及复制图片到新位置。

摘要由CSDN通过智能技术生成

import os
import cv2
import random
import shutil
# 获取文件夹名称
path = r't'
dirnames = [f for f in os.listdir(path) if os.path.isdir(path + '\\' + f)]
print(dirnames)
# 遍历文件夹
for dir_name in dirnames:
    dir_path = path + '\\' + dir_name
    print(dir_path)
    # 创建新的文件夹
    target_path = r'f2' + '\\' + dir_name + '\\'
    if not os.path.exists(target_path):
        os.makedirs(target_path)
    pirpath = os.listdir(dir_path)
    # 随机选取函数
    sample = random.sample(pirpath, 2)  # 随机选取一定数量的样本图片，这里选2个
    print(sample)
    # 保存图片到目标文件夹
    for name in sample:
        print(name)
        shutil.copyfile(os.path.join(dir_path,name), os.path.join(target_path,name))
        # copyfile复制的是文件

遇到最大的问题就是一开始在复制随机选取的图片到其他文件夹时，不能将图片路径和名字拼接在一起，后来用os.path.join将文件夹名字和图片名字拼接到一起了。