Tensorflow2.0 对自己的图片数据集进行分类

文章目录

  • 项目介绍
    • 数据集
    • 网络模型
  • 代码实现
    • 1、导入需要的库
    • 2、定义图像加载和预处理函数
    • 3、定义构造 Dataset 数据集函数
    • 4、构造 Dataset 数据集
    • 5、构建网络
    • 6、初始化优化器和损失函数
    • 7、定义损失函数
    • 8、定义梯度下降函数
    • 9、保留 Checkpoint 文件
    • 10、训练过程
    • 11、保存模型到 .h5 文件中
    • 12、绘制准确率曲线

你可能感兴趣的:(python,深度学习,PHM项目实战--建模篇,tensorflow,python,深度学习)