![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=1739153458-cjGZqOPlyRSnLtYLaS7kgsY6jGFJ3xRo-0-2ab62349bfee8289a6953fc505c628fb)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P45_48329.jpg?sign=1739153458-wF3oV3XkTUUiLCRODTdMqaHsCQUD7P7G-0-d6aebf2bdb3fe1088914790d7ddd6cd7)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P46_48330.jpg?sign=1739153458-1BLhksVTe5lMLq54s3Z4is2ssrUEKsUU-0-8c338fed254329fbb53ab5987ac760b6)
在当前文件夹中新建程序train.py,负责模型的训练。训练逻辑如程序源码P1.3所示。
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P46_48331.jpg?sign=1739153458-WYYl7U4bi7WtRQJXn6aHCsYAIbuBGFEV-0-6afb815e114aae851bb81abce7d120b9)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P47_48332.jpg?sign=1739153458-11KMBVJ3Yb2gNcZPVaMXu81wbbZuUOfa-0-d7d8d3d934e300c757ad1da2aaf5c75b)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P48_48333.jpg?sign=1739153458-yTSwZyEl9dx69OdlFQfRL4rj3q1QJ25A-0-e8d1c764a1dafb951ae26d53b7fae81f)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P49_48334.jpg?sign=1739153458-Kxj8VCsvWpyZJtG3k0dBtIHddTHXstu0-0-96a81028b980dccba21f00010d8b3c0c)
运行程序train.py,开始模型训练过程,模型训练细节、程序设计细节、运行测试结果及详细解析参见本节微课视频。