融合MobileNetV3特征的结构化剪枝方法

刘宇, 雷雪梅

A Structured Pruning Method Integrating Characteristics of MobileNetV3

LIU Yu, LEI Xuemei

表5 剪枝前后模型通道数对比

Tab.5 Comparison of model channels before and after pruning

输入尺寸 模块 模块中 通道数 模块中通道 数(剪枝后) 输出 通道数 激励模块 激活函数 步长 224×224×3 conv2d — — 16 — HS 2 112×112×16 bneck, 3×3 16 9 16 — RE 1 112×112×16 bneck, 3×3 64 49 24 — RE 2 56×56×24 bneck, 3×3 72 42 24 — RE 1 56×56×24 bneck, 5×5 72 72 40 √ RE 2 28×28×40 bneck, 5×5 120 102 40 √ RE 1 28×28×40 bneck, 5×5 120 89 40 √ RE 1 28×28×40 bneck, 3×3 240 223 80 — HS 2 14×14×80 bneck, 3×3 200 144 80 — HS 1 14×14×80 bneck, 3×3 184 139 80 — HS 1 14×14×80 bneck, 3×3 184 112 80 — HS 1 14×14×80 bneck, 3×3 480 209 112 √ HS 1 14×14×112 bneck, 3×3 672 38 112 √ HS 1 14×14×112 bneck, 5×5 672 540 160 √ HS 2 7×7×160 bneck, 5×5 960 484 160 √ HS 1 7×7×160 bneck, 5×5 960 255 160 √ HS 1 7×7×160 conv2d, 1×1 — — 960 — HS 1 7×7×960 pool, 7×7 — — — — — 1 1×1×960 conv2d, 1×1, NBN — — 1 280 — HS 1 1×1×1280 conv2d, 1×1, NBN — — q — — 1