文章目录
- 项目介绍
-
- 数据集
- 网络模型
- 代码实现
-
- 1、导入需要的库
- 2、定义图像加载和预处理函数
- 3、定义构造 Dataset 数据集函数
- 4、构造 Dataset 数据集
- 5、构建网络
- 6、初始化优化器和损失函数
- 7、定义损失函数
- 8、定义梯度下降函数
- 9、保留 Checkpoint 文件
- 10、训练过程
- 11、保存模型到 .h5 文件中
- 12、绘制准确率曲线
- 13、绘制损失曲线
项目介绍
数据集
三分类问题，共有几百张图片，每张图片都是文件类型为 bmp 的彩色图片，按照 8：2 的比例被分成训练集和测试集，图片所在位置如下图所示：