目录
图像分类领域经典的3种算法:Alex网络、VGG网络、ResNet网络。
序言
※本项目是与组员共同创作非个人独立完成
本文主要介绍使用python搭建:一个使用了ResNet的水果图像分类识别系统。
使用ResNet进行图像分类的相关代码。
-
数据预处理,生成用于输入TensorFlow模型的TFRecord的数据。
-
模型构建及训练,使用tensorflow.keras构建深度残差网络。
-
预测水果分类并进行模型评估。
一、数据预处理
-
读取数据集:本次参与训练的数据集共81个类别,数据集格式为ImageNet数据集格式
-
处理数据: