视频来源:
https://www.bilibili.com/video/av67348232?from=search&seid=10221504884724623592
常规卷积
相信接触过卷积神经网络的都知道常规卷积的操作
我们通过N个DkDk大小的卷积核
卷积出来的结果
设为DnDn*N
现在我们来计算一下常规卷积的计算开销
一个卷积核做一次卷积操作需要的开销为
Dk*Dk*M
而每个卷积核完整地卷积完一次所需开销为
Dg*Dg*Dk*Dk*M
这里我们假设卷积出来的结果长宽为Dg
然后我们使用了N个卷积核来进行卷积
所以总计算开销为
N*Dg*Dg*Dk*Dk*M
DepthWise卷积
下面我们来看一下Depthwise卷积
常规的卷积中,每个卷积核都对每个通道进行了一次计算