深度学习数据集划分

最新推荐文章于 2024-05-06 10:23:59 发布

想学算法的蒟蒻

最新推荐文章于 2024-05-06 10:23:59 发布

阅读量599

点赞数

分类专栏： python 文章标签：深度学习 python 人工智能

本文链接：https://blog.csdn.net/zhb3439/article/details/127940059

版权

python 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

深度学习数据集划分Python

import os
from shutil import copy
import random
file_path = 'dataset/val_masks'#源图像位置


images = os.listdir(file_path)  # iamges 列表存储了该目录下所有图像的名称
num = len(images)
print(num)
eval_index = images[:1415] #按照8:2划分训练集测试集
for img in images:
    if img in eval_index:
        image_path = file_path +'/'+img
        copy(image_path, 'dataset/train_data/masks')  # 将选中的图像复制到新路径
    else:
        image_path = file_path +"/"+ img
        copy(image_path, 'dataset/test_data/masks')
print('done')