Python：创建数据集目录，新建文件夹

在划分数据集时，需要按照数据集的格式来创建目录，最后创建出来的目录形式如下：

在创建文件夹时，要先判断文件夹是否存在。用python里的函数isdir()来判断文件夹是否存在，如果不存在，才需要创建，用python里的函数makedirs()来创建文件夹。

按照上述目录，创建的代码如下：

#Dir是数据集根目录
def create_datasetdir(dir):
    if not os.path.isdir(dir):
        os.makedirs(dir)#如果根目录不存在则创建
    train_path = os.path.join(dir, "train")
    if not os.path.isdir(train_path):
        os.makedirs(train_path)#如果train目录不在，则创建
    train_image_path = os.path.join(train_path, "images")
    if not os.path.isdir(train_image_path):
        os.makedirs(train_image_path)
    train_label_path = os.path.join(train_path, "labels")
    if not os.path.isdir(train_label_path):
        os.makedirs(train_label_path)
    val_path = os.path.join(dir, "val")
    if not os.path.isdir(val_path):
        os.makedirs(val_path)
    val_image_path = os.path.join(val_path, "images")
    if not os.path.isdir(val_image_path):
        os.makedirs(val_image_path)
    val_label_path = os.path.join(val_path, "labels")
    if not os.path.isdir(val_label_path):
        os.makedirs(val_label_path)

测试代码如下：

create_datasetdir(r"E:\fishdataset")

本文最前面的图即是通过测试代码生成的文件夹。

也可直接在E:\fishdataset下查找到所创建的文件夹。