文章目录
- 项目介绍
- 划分数据集
-
- 1、在 cocofisher_pic2 目录下创建两个文件夹
- 2、将原来 cocofisher_pic2 文件夹中的子文件夹分别复制到这两个个文件夹中
- 3、在 python 中指定所有路径
- 4、将这两个文件夹中所有子文件夹中的图片删除
- 5、将 cocofisher_pic2 目录下子文件中的原始图片两个文件夹中
- 导入图片
-
- 1、构建图片生成器
- 2、将图片输入图片生成器
- 训练模型
-
- 1、构建模型
- 2、训练模型
- 3、测试模型
项目介绍
在下载完一个数据集后，很多朋友会为如何划分出训练集、验证集和测试集而烦恼，这篇文章将详细介绍如何划分数据集并将其用于模型训练。
首先，我们的原始数据集如下图所示：