文章目录
项目介绍
数据集
三分类问题,共有几百张图片,每张图片都是文件类型为 bmp 的彩色图片,按照 8:2 的比例被分成训练集和测试集,图片所在位置如下图所示:
【注】如果数据集没有分成训练集和测试集,可以参考将自己的图片数据集分成训练集、验证集和测试集这篇文章。
网络模型
我们使用的网络模型是 MobileNet,可以直接通过 Tensorflow2.0 中提供的迁移学习函数调用。
代码实现
1、导入需要的库
import tensorflow as tf
import