Python创建数据集目录,新建文件夹
在划分数据集时,需要按照数据集的格式来创建目录,最后创建出来的目录形式如下:
在创建文件夹时,要先判断文件夹是否存在。用python里的函数isdir()来判断文件夹是否存在,如果不存在,才需要创建,用python里的函数makedirs()来创建文件夹。
按照上述目录,创建的代码如下:
#Dir是数据集根目录
def create_datasetdir(dir):
if not os.path.isdir(dir):
os.makedirs(dir)#如果根目录不存在则创建
train_path &#