当前位置：首页 > article >正文

标准卷积（Standard Convolution）

article 2025/3/12 10:07:15

标准卷积的基础操作图解：

卷积之后尺寸公式：

输入尺寸：W×H
卷积核尺寸：Fw×Fh
填充大小：P
步长：S

输出尺寸 Wout×Hout可以通过以下公式计算：

$W_{out}=[\frac{W+2P-F_{w}}{S}]+1$

$H_{out}=[\frac{W+2P-F_{h}}{S}]+1$

其中[x]表示向下取整。

实例：

输入图像尺寸：5x5
卷积核尺寸：3x3
填充：1
步长：1

输出尺寸：

$W_{out}=[\frac{5+2\cdot 1-3}{1}]+1=5$

$H_{out}=[\frac{5+2\cdot 1-3}{1}]+1=5$

所以输出特征图尺寸为5×5.

代码实例：

import torch
import torch.nn as nn

# 创建随机输入张量 (batch_size, channels, height, width)
input_tensor = torch.randn(1, 1, 5, 5)  # 1个样本，1个通道，5x5大小的图像

# 定义卷积层
conv_layer = nn.Conv2d(in_channels=1, out_channels=1, kernel_size=3, padding=1, stride=1)

# 执行卷积操作
output_tensor = conv_layer(input_tensor)

# 打印输出特征图的尺寸
print("Output tensor shape:", output_tensor.shape)

输出：