本例提取了植物幼苗数据集中的部分数据做数据集，数据集共有12种类别，演示如何使用pytorch版本的MobileNetV1图像分类模型实现分类任务。
通过本文你和学到：
1、如何自定义MobileNetV1模型。
2、如何自定义数据集加载方式？
3、如何使用Cutout数据增强？
4、如何使用Mixup数据增强。
5、如何实现训练和验证。
6、预测的两种写法。
2022-02-01 21:06:14
948.38MB