class DWConv(Conv): # Depth-wise convolution class def __init__(self, c1, c2, k=1, s=1, act=True): # ch_in, ch_out, kernel, stride, padding, groups super().__init__(c1, c2, k, s, g=math.gcd(c1, c2), act=act)
时间: 2023-06-20 10:03:50 浏览: 289
Going deeper with convolutions
5星 · 资源好评率100%
这是一个深度可分离卷积(Depth-wise Separable Convolution)的实现类,它继承了一个普通卷积的类 Conv。深度可分离卷积是一种轻量级卷积操作,通常用于深度学习中的计算机视觉任务。相比于普通卷积,深度可分离卷积可以将计算量大幅减少,从而加速模型的训练和推理。
具体地,深度可分离卷积包含两个步骤:深度卷积和逐点卷积。深度卷积是指对每个输入通道单独进行卷积操作,得到一个与输入通道数相同的中间特征图。逐点卷积是指对中间特征图的每个像素点单独进行卷积操作,得到最终的输出特征图。这种卷积方法可以大幅度减少计算量,同时还能有效地提高模型的表现力。
阅读全文