【存档】转置卷积&膨胀卷积

深入理解转置卷积与膨胀卷积在PyTorch中的应用

最新推荐文章于 2025-04-17 18:06:11 发布

原创

最新推荐文章于 2025-04-17 18:06:11 发布 · 1k 阅读

13 ·

CC 4.0 BY-SA版权

文章标签：

#python #机器学习 #卷积神经网络

本文介绍了转置卷积（TransposedConvolution）及其在PyTorch中的实现（nn.ConvTranspose2d），并探讨了膨胀卷积（DilationConvolution）的作用，尤其是在增大感受野和HybridDilatedConvolution(HDC)中的应用。文中还讨论了不同膨胀系数对堆叠效果的影响。

A. Transposed Convolution

参考：转置卷积
 转置卷积

上采样，实现从低维到高维
属于卷积，但不等于逆卷积

Pytorch实现nn.ConvTranspose2d

用torch.nn中的nn.ConvTranspose2d()函数，参数如下（dilation为膨胀元素一般为1）
在这里插入图片描述
输出特征图尺寸计算公式：

具体实现

import torch
import torch.nn as nn

def transposed_conv_official():
    feature_map = torch.as_tensor([[1, 0],
                                   [2, 1]], dtype=torch.float32).reshape([1, 1, 2, 2