Author:XuLiu
Time:20211123
Fuction:数据集划分,但是有区别的是将文件夹里的图片划分到train和test中
而且train和test是没有重合的
输入:
输出:根据设定的比例切分到train和test
代码如下:
import os
import random
import shutil
from shutil import copy2
"""os.listdir会将文件夹下的文件名集合成一个列表并返回"""
def getDir(filepath):
pathlist = os.listdir(filepath)
return pathlist
"""如果不存在文件夹,就建一个文件夹"""
def makesDir(filepath):
if not os.path.exists(filepath):