前言
虽然这个主要是语义分割的轻量级网络模型,但是模块的设计还是很有参考意义的。
ESPNetV1
论文:https://arxiv.org/abs/1803.06815v2
代码:https://github.com/sacmehta/ESPNet
创新点
这篇文章主要创新点也就是ESP模块的设计,将标准的卷积分解为2个部分
- point-wise convolutions 逐点卷积
- spatial pyramid of dilated convolutions 空间金字塔型膨胀卷积
主要的模块设计的策略为 reduce-split-transform-merge
,如下图所示
对于一个的标准卷积核,ESP模块将其分解为以下三步:
- Step-1: Reduce(Point-wise convolutions):d个1x1xM的卷积核,将M维的输入feature map降至d维;
- Step-2: Split & Transform(Spatial pyramid of dilate