![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=1738881947-IGGMitqbAvXlLm4TkMabHmLsF8rYkOmm-0-57ca2782978d635dc809eca5364e2dd1)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P45_48329.jpg?sign=1738881947-DkcXO0Kmlj86J2QypbpSBhh5bHqxG9sn-0-cc7eecd58a1160de53c38cd3b5c02ac9)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P46_48330.jpg?sign=1738881947-ZadWp8GanINlIZBjyrOjKLB4G9qI7eq5-0-77e1b2001fbfae8c51d82da2236941e6)
在当前文件夹中新建程序train.py,负责模型的训练。训练逻辑如程序源码P1.3所示。
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P46_48331.jpg?sign=1738881947-Yz5OujrX394rdylKju72QDe5OTMqAlDk-0-5648e4f0bba19c8b3d8ebd17252865cd)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P47_48332.jpg?sign=1738881947-zLs6II52hZ03xdMJQTTec9HyZKQeEGRv-0-095debd709a7b04ba2cf5f179dbb4378)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P48_48333.jpg?sign=1738881947-XeX7ce7xS1IIcUvCeTlBaMdjybZYkl0v-0-54636d1a57d31ddd673ffacdb3d50da1)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P49_48334.jpg?sign=1738881947-A8IHOtZO78QDmfoDry5YbVbpHb2TMC3d-0-3a74432dff7906b69ed74cc0f2d78982)
运行程序train.py,开始模型训练过程,模型训练细节、程序设计细节、运行测试结果及详细解析参见本节微课视频。