- TensorFlow+Android经典模型从理论到实战(微课视频版)
- 董相志 曲海平 董飞桐编著
- 383字
- 2024-12-27 22:42:45
1.6 EfficientNetV2建模
参照EfficientNetV2论文公布的源码,本节完成EfficientNetV2-S/M/L三个模型的编程。官方程序用列表描述模型的参数结构,以EfficientNetV2-S模型为例,其结构描述如下。
v2_s_block = [ 'r2_k3_s1_e1_i24_o24_c1' 'r4_k3_s2_e4_i24_o48_c1' 'r4_k3_s2_e4_i48_o64_c1' 'r6_k3_s2_e4_i64_o128_se0 25' 'r9_k3_s1_e6_i128_o160_se0 25' 'r15_k3_s2_e6_i160_o256_se0 25' ]
参照表1.9对EfficientNetV2-S模型的定义,v2_s_block中的参数符号解读如下。
r:当前Stage中Operator重复的次数repeats。
k:卷积核尺寸kernel_size。
s:步长stride。
e:升维倍率因子expansion ratio。
i:当前Stage的输入通道数input channels。
o:当前Stage的输出通道数output channels。
c:卷积类型,0表示Fused-MBConv,1表示MBConv。
se:表示SE模块第一个全连接层的通道压缩因子se_ratio。
打开PyCharm,新建项目TensorFlow_to_Android,按照本节视频教程提示,配置项目虚拟环境,安装TensorFlow和相关库。新建文件夹EfficientNetV2,在其中创建model.py程序,完成程序源码P1.1的编程与测试工作。
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P36_48320.jpg?sign=1739086082-PCPnl1u8vgFDtbSFc06kclTzpuJYVhfI-0-9d650bfad554ca40474af617ecae308a)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P37_48321.jpg?sign=1739086082-NZZbrwHgntzDFPTuk4xIUCahMKvSK9vx-0-7c73a439dd139e2579131cd26753177f)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P38_48322.jpg?sign=1739086082-NAanl21eBOhGWrhP98u92RiB4cFkgY7G-0-cfdb4714cacca7dd8e37d1b0119b36fa)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P39_48323.jpg?sign=1739086082-QJ8k0ZOuUAUzsuRHhWZMFuedZXg1BQN3-0-e99351a853c21032be4db96afe98fad1)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P40_48324.jpg?sign=1739086082-it2b7wZ8o1aFmTLFIw5ga5Fu4bUtkR6A-0-860758aef57da777fd176395a995b234)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P41_48325.jpg?sign=1739086082-vm4FdSOWewlUnOMP9IoU7WEQyFnuIU0l-0-68abf06b9e2d2aafe439593c398f67b9)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P42_48326.jpg?sign=1739086082-uBhey6bKZyg7TUhjW2tJR8z1KbYWG9Il-0-9e8542b25835a5013b3cd857ecb1a06f)
![](https://epubservercos.yuewen.com/A52FB2/31398333104740106/epubprivate/OEBPS/Images/Figure-P43_48327.jpg?sign=1739086082-j6HcLlrdEobprOgU9uZKUd8yUvJt7IW5-0-a41c272396f1a53c3f303c808a849921)
程序model.py编码有些长,其中包含了EfficientNetV2-S/M/L三种模型的实现逻辑,该逻辑极其容易扩展到其他模型,包括EfficientNetV1的系列模型。
程序model.py编码优雅,面向对象的结构化设计,层层递进,值得学习和模仿。
程序设计细节、运行测试结果及详细解析参见本节微课视频。