![TensorFlow+Android经典模型从理论到实战(微课视频版)](https://wfqqreader-1252317822.image.myqcloud.com/cover/815/52842815/b_52842815.jpg)
上QQ阅读APP看书,第一时间看更新
1.7 EfficientNetV2训练
1.3节中介绍过包含104种花朵的数据集,为了便于在普通PC的主机上演示EfficientNetV2模型训练过程,本节暂且采用包含5种花朵的小型数据集作为教学演示。随后转到Kaggle平台上完成针对104种花朵数据集的模型训练。
接1.6节的工作,在当前文件夹中新建程序utils.py,定义数据集预处理函数,包括随机划分、数据增强等,如程序源码P1.2所示。源码解析参见本节视频讲解。
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P44_48328.jpg?sign=1739026546-lm0nDUq3P3GWDSP5JqLyioaoPHRJuzHH-0-1c2ba3f8dc45cd46595b4566694eb376)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P45_48329.jpg?sign=1739026546-U78zYnX3J9rzWAl9fb5etGscXuX73UOI-0-f15486abe39c95563cd495f2d9f4e09d)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P46_48330.jpg?sign=1739026546-T4zX17W4vTbEn4dnU1kkpyPYjnWkwFBW-0-866af64bb3a57cb17abae8fb82083eaa)
在当前文件夹中新建程序train.py,负责模型的训练。训练逻辑如程序源码P1.3所示。
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P46_48331.jpg?sign=1739026546-KjEdyAcwCIACzC7AWSvtrhLUfSRaPd0w-0-9e7602ce7256455fb66b750d4a1149ee)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P47_48332.jpg?sign=1739026546-q9KRDKK9Xp1G0DPeEr3mJxdDY3zDZQgr-0-67ea28679a810e88878a01ddfa557736)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P48_48333.jpg?sign=1739026546-WQYs6NpkMoHil5WjnT3PAsWTgHdlq72Q-0-5448f8c415a50ec83b2d5faa5aa755d8)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P49_48334.jpg?sign=1739026546-mSrpXr3J0SnlV1UhxD3ieVh3BT5ZAEfU-0-4679ba7ef16d9ee06a4253a85948cfb7)
运行程序train.py,开始模型训练过程,模型训练细节、程序设计细节、运行测试结果及详细解析参见本节微课视频。