从文件夹大量图片中随机抽取图片当训练集和测试集

最新推荐文章于 2024-06-26 12:51:13 发布

啥都不会的fw

最新推荐文章于 2024-06-26 12:51:13 发布

阅读量1.2k

点赞数 1

文章标签： python

本文链接：https://blog.csdn.net/qq_40860979/article/details/124197258

版权

@TOC从文件夹大量图片中随机抽取图片当训练集和测试集

import os, random, shutil


def moveFile(fileDir):
    # pathDir = os.listdir(fileDir)
    # filenumber = len(pathDir)
    # rate = 0.001
    # picknumber = int(filenumber * rate)
    # sample = random.sample(pathDir, picknumber)
    # print(sample)
    # for name in sample:
    #     shutil.move(fileDir + name, tarDir + name)
    # return
    #
    # pathDir = os.listdir(tarDir)
    # result = os.listdir(fileDir)
    # for list in result:
    #     if list in pathDir:
    #         shutil.move(fileDir + list, newdir + list)

    hr_img = os.listdir(fileDir)
    filenumber = len(hr_img)
    picknumber = 1000
    sample = random.sample(hr_img, picknumber)


    for name in sample:
        shutil.move(fileDir + name, tarDir + name)

    result = os.listdir(tarDir)
    pathDir = os.listdir(hr_hazyDir)
    for list in result:
        if list in pathDir:
            shutil.move(hr_hazyDir + list, tarhazydir + list)


if __name__ == '__main__':
    hrdir = "sidd320/HR/" #源文件夹
    hr_hazyDir = "sidd320/HR_hazy/"#源文件夹
    tarDir = 'sidd320test/HR/' #目标文件夹
    tarhazydir = 'sidd320test/HR_hazy/'#目标文件夹
    moveFile(hrdir)