2021-08-23

Posted 2021-09-09 奔跑的小豆芽

python 数据集中随机选择作为测试集、验证集

一、功能

从已有的数据集中随机选择出若干组数据作为测试集或者测试集。

二、数据

数据集由三种数据组成，分别是：前期数据（tif）、后期数据(tif)、标签图（png），所以数据集由三个文件夹组成。

二、直接上代码

##深度学习过程中，需要制作训练集和验证集、测试集。

import os, random, shutil

def moveFile(fileDir):
    pathDir = os.listdir(fileDir)  # 取图片的原始路径