python實現從一個文件夾下隨機抽取一定數量的圖片並移動到另一個文件夾

由於制作訓練集和驗證集、測試集的需要，我需要從原始數據集中，隨機取出（剪切）一定數量的圖片。手動挑選畢竟不能保證隨機性，且費時費力。故參考了這位博主的方法。

##深度學習過程中，需要制作訓練集和驗證集、測試集。

import os, random, shutil
def moveFile(fileDir):
        pathDir = os.listdir(fileDir)    #取圖片的原始路徑
        filenumber=len(pathDir)
        rate=0.1    #自定義抽取圖片的比例，比方說100張抽10張，那就是0.1
        picknumber=int(filenumber*rate) #按照rate比例從文件夾中取一定數量圖片
        sample = random.sample(pathDir, picknumber)  #隨機選取picknumber數量的樣本圖片
        print (sample)
        for name in sample:
                shutil.move(fileDir+name, tarDir+name)
        return

if __name__ == '__main__':
    fileDir = "./source/"    #源圖片文件夾路徑
    tarDir = './result/'    #移動到新的文件夾路徑
    moveFile(fileDir)   

如果是要復制文件可參考以下完整代碼

import os, random, shutil
def copyFile(fileDir):
        pathDir = os.listdir(fileDir)    #取圖片的原始路徑
        filenumber=len(pathDir)
        rate=0.01    #自定義抽取圖片的比例，比方說100張抽10張，那就是0.1
        picknumber=int(filenumber*rate) #按照rate比例從文件夾中取一定數量圖片
        sample = random.sample(pathDir, picknumber)  #隨機選取picknumber數量的樣本圖片
        print (sample)
        for name in sample:
                shutil.copy(fileDir+name, tarDir+name)
        return

if __name__ == '__main__':
    fileDir = "./source/"    #源圖片文件夾路徑
    tarDir = './result/'    #移動到新的文件夾路徑
    copyFile(fileDir)