本例提取了植物幼苗数据集中的部分数据做数据集,数据集共有12种类别,今天我和大家一起实现tensorflow2.X版本图像分类任务,分类的模型使用MobileNetV2。本文实现的算法有一下几个特点:
1、自定义了图片加载方式,更加灵活高效,不用将图片一次性加载到内存中,节省内存,适合大规模数据集。
2、加载模型的预训练权重,训练时间更短。
3、数据增强选用albumentations。
文章链接:https://blog.csdn.net/hhhhhhhhhhwwwwwwwwww/article/details/122768747
2022-02-02 17:05:56
937.01MB
图像分类