深度学习论文中经常出现的Depthwise卷积核Pointwise卷积是个啥？

一头小学牲

于 2023-04-05 14:39:55 发布

阅读量154

点赞数 1

分类专栏：学习笔记文章标签：深度学习人工智能计算机视觉

本文链接：https://blog.csdn.net/weixin_42699538/article/details/129970521

版权

学习笔记专栏收录该内容

11 篇文章 2 订阅

订阅专栏

1. Depthwise Convolution

一图胜千言，图来自PapersWithCode，侵删
请添加图片描述
pytorch 代码实现：

conv_depthwise = torch.nn.Conv2d(
    in_channels,
    in_channels, # 注意，输出通道数和输入通道数是相等的，都是in_channels
    kernel_size=kernel_size,
    stride=stride,
    padding=padding,
    bias=False,        # 是否使用bias看自己需求
    groups=in_channels # 指定这个groups参数是关键
)

2. Pointwise Convolution

请添加图片描述

conv_pointwise = torch.nn.Conv2d(
	  in_channels,
      out_channels,
      kernel_size=1,
      stride=1,
      padding=0,
      bias=False  # 是否使用bias看自己需求
)