文章目录
- 8.1 ShuffleNet v1 v2理论讲解
- ShuffleNet v1
- ShuffleNet v2
- 8.2 使用Pytorch搭建ShuffleNet
8.1 ShuffleNet v1 v2理论讲解
ShuffleNet v1
论文链接:https://readpaper.com/paper/2963125010
Channel shuffle
相关链接:深度学习-【图像分类】学习笔记 6ResNet
group conv占据的时间很少,大部分卷积都被1 × 1的PW Conv占据了。
因此将所有的 1 × 1 的Conv都换成了 1 × 1 GConv。
如何搭建 Shuffle Net?
图中的g就是group。
核心思想:
- 在Group Conv后加上一个Channel Shuffle模块。
- 对于Block将其中的1×1卷积改成Group Conv。
ShuffleNet v2
论文链接:https://readpaper.com/paper/2883780447
比较硬核,建议直接去看原论文。
作者提出:
作者提出了设计高效网络的4条建议——
分条来看:
保持FLOPs不变,表示 B = hwc1c2是个常数。