在划分数据集时,需要按照数据集的格式来创建目录,最后创建出来的目录形式如下: 在创建文件夹时,要先判断文件夹是否存在。用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下查找到所创建的文件夹。
|