读取文件夹下的所有图片,随机提取一定数量并保存至另一文件夹

直接贴出代码(Python)

import os
import random
import shutil
import skimage.io as io

def copyFile(fileDir,tarDir):
    
    pathDir = os.listdir(fileDir)
    for filename in pathDir:
        print (filename)


    coll = io.ImageCollection(str)
    print(len(coll)) #打印图片数量
    num = int ((2*len(coll))/10)
    print(num)
    sample = random.sample(pathDir,num)
    
    for name in sample:
        shutil.copyfile(fileDir+name,tarDir+name)



if _name_ == '_main_':
   fileDir = "       " #填写要读取图片文件夹的路径
   tarDir = "        " #填写保存随机读取图片文件夹的路径
   str = 'fileDir*.jpg' # fileDir的路径+*.jpg表示文件下的所有jpg图片
   copyFile(fileDir,tarDir)

修改相关路径即可。

 

 

 

 

你可能感兴趣的:(读取文件夹下的所有图片,随机提取一定数量并保存至另一文件夹)