![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=1739040636-tpI4ftAIpVte6KHXPkGnXZSQwiOJIRDg-0-8381bfd04d30c870ef370eebf2227dd9)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P45_48329.jpg?sign=1739040636-muKUsiqpQqoJmVSSrbt8AKv5aX8hPEO4-0-dd07c4e26ffb4ae0cbb9b23302b1c54d)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P46_48330.jpg?sign=1739040636-N0FUNsCO7EkB5k2TliTvZVmtxstrXUxe-0-c61b58b6184b8102667b7c23d6bc297a)
在当前文件夹中新建程序train.py,负责模型的训练。训练逻辑如程序源码P1.3所示。
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P46_48331.jpg?sign=1739040636-zjBESuQF7owpSDZF4VYp7AtXoCizvi4p-0-33eabcff0ed174b60fe568afa271ca5f)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P47_48332.jpg?sign=1739040636-hTsEnVda8PwZwZFzVVngE9FZFpPcf0Xl-0-b90bcdd42c1bbb85440a88681c272ba5)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P48_48333.jpg?sign=1739040636-PqSGASppt7yE4YOxYgOBAiCBz27HivUC-0-3c3a1d18ff827ef9d5e544f8aaebe212)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P49_48334.jpg?sign=1739040636-2XOQarOZ7t9e38g5IZHPXgO2xlhRQzGA-0-f1408a5386a14c225dc6bf55d0c6e3da)
运行程序train.py,开始模型训练过程,模型训练细节、程序设计细节、运行测试结果及详细解析参见本节微课视频。