随机抽取一个文件夹中的部分图片并复制到另一个文件夹

delight0123

已于 2023-12-28 16:42:44 修改

阅读量722

点赞数

分类专栏： Python 文章标签： python

于 2022-11-13 00:00:13 首次发布

本文链接：https://blog.csdn.net/qq_54444533/article/details/127827438

版权

图片处理文件操作 Python编程随机抽取数据集管理

关键词由CSDN通过智能技术生成

Python 专栏收录该内容

13 篇文章 0 订阅

订阅专栏

import os
import random
import shutil

def copyFile(fileDir):
    
    pathDir = os.listdir(fileDir)  
    filenumber = len(pathDir)
    rate = 0.5  # 自定义抽取图片的比例，此处随机抽取一半
    picknumber = int(filenumber*rate)  # 按照rate比例从文件夹中取一定数量图片
    sample = random.sample(pathDir, picknumber)  # 随机选取picknumber数量的样本图片
    print(sample)
    for name in sample:
        shutil.copy(fileDir+name, tarDir+name)
    return

fileDir = "C:/Users/22720/Desktop/mnist (2)/images60000/"   #源文件夹
tarDir = "C:/Users/22720/Desktop/images30000/"   #目标文件夹
copyFile(fileDir)

随机抽取源文件夹中的图片并移动到目标文件夹中。