深度学习卷积基础知识

曙光_deeplove

于 2022-09-06 14:36:08 发布

阅读量468

点赞数

分类专栏：深度学习文章标签：深度学习 python cnn

本文链接：https://blog.csdn.net/yangjinyi1314/article/details/126724409

版权

深度学习专栏收录该内容

46 篇文章 7 订阅

订阅专栏

一、

这样就没错了，是4组 3*3 * 3 的核在计算

二、

https://www.jianshu.com/p/16a4e9831a59

class DWConv(nn.Module):
    def __init__(self, in_plane, out_plane):
        super(DWConv, self).__init__()
        self.depth_conv = nn.Conv2d(in_channels=in_plane,
                                    out_channels=in_plane,
                                    kernel_size=3,
                                    stride=1,
                                    padding=1,
                                    groups=in_plane)
        self.point_conv = nn.Conv2d(in_channels=in_plane,
                                    out_channels=out_plane,
                                    kernel_size=1,
                                    stride=1,
                                    padding=0,
                                    groups=1)
    def forward(self, x):
        x = self.depth_conv(x)
        x = self.point_conv(x)
        return x

这个深度卷积就groups不一样，我估计就是控制如何卷积的
groups等于输入通道就是让卷积核每层分别去卷积，最终结果不相加。