Motivation
在做图像退化的时候发现pytorch早期版本不支持same padding(新版本是有padding = 'same'
这个选项的,但是本人试了也没成功,暂时没找到原因)。于是去查有没有办法自己实现一下same padding。
torch.nn.functional.conv2d 详解
官方文档写的非常清楚,直接贴上来
实现方法
由于F.conv2d
的padding最多只支持两个参数(对称填充),有些输出大小没办法实现。因此可以通过F.pad()
先做填充,再用F.conv2d
做卷积。
import torch.nn.functional as F
import torch.nn as nn
filters = torch.randn(1, 1, 8, 15)
inputs = torch.randn(16, 1, 256, 256)
# 手动填充
inputs = F.pad(inputs, (7, 7, 4, 3))
out = F.conv2d(inputs, filters)
print(out.shape)
输出:
torch.Size([16, 1, 256, 256])