深度可分离卷积（Depth-wise Separabel Convolution）

最新推荐文章于 2024-04-14 03:15:56 发布

原创

最新推荐文章于 2024-04-14 03:15:56 发布 · 1.4k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #深度学习

深度可分离卷积

深度可分离卷积 = depth-wise conv + point-wise conv。
depth-wise conv：输入特征图每个通道作为一个group，用一个kernel处理。
point-wise conv：kernel size =1 的普通卷积。

import torch
import torch.nn as nn

stride = 1
kernel_size = 3
in_ch = 10
out_ch = 3

conv_normal = nn.Conv2d(in_channels=in_ch,
                        out_channels=out_ch,
                        kernel_size=kernel_size,
                        stride=stride)

input = torch