数据集自动分类训练集和验证集代码

最新推荐文章于 2024-07-25 16:46:55 发布

Gewenfei_1

最新推荐文章于 2024-07-25 16:46:55 发布

阅读量2.9k

点赞数

分类专栏：深度学习图像文章标签：深度学习数据集预处理

本文链接：https://blog.csdn.net/Gewenfei_1/article/details/81214728

版权

本文介绍如何将数据集自动分类为训练集和验证集，特别是对于具有正负样本的分类任务，如猫狗大战数据集。通过随机方式，将原始数据目录结构调整为包含训练和验证子目录，确保模型训练过程中的数据分布随机性。

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

针对类似于猫狗大战的数据集，拿到数据集的目录结构为：
这里写图片描述

在label_00文件夹下是正样本，label_11下是负样本。此时如果想要分离验证集并且带有随机性，代码如下：

# -*- coding: utf-8 -*-
import shutil
import keras
import numpy as np
from keras.models import Sequential
from keras.initializers import he_normal
from keras.layers.normalization import BatchNormalization
from keras.callbacks import ReduceLROnPlateau
from keras.callbacks import ModelCheckpoint
from keras.layers import</