python划分train val test

最新推荐文章于 2024-08-17 08:28:21 发布

qq_40770036

最新推荐文章于 2024-08-17 08:28:21 发布

阅读量652

点赞数

文章标签： python

本文链接：https://blog.csdn.net/qq_40770036/article/details/105710583

版权

这篇博客分享了如何处理文件夹中的图片数据集，使用Python代码将数据划分为训练集、验证集和测试集，适用于机器学习任务。

摘要由CSDN通过智能技术生成

最近一个任务需要用到数据集，但是下载到的数据集是文件夹/图片形式的，所以需要对数据进行划分，代码如下


import os
from shutil import copyfile
def mkdir(path):
    folder = os.path.exists(path)

    if not folder:  # 判断是否存在文件夹如果不存在则创建为文件夹
        os.makedirs(path)  # makedirs 创建文件时如果路径不存在会创建这个路径
        print
        "---  new folder...  ---"
        print
        "---  OK  ---"