CNN 卷积输出尺寸计算器
根据输入尺寸、卷积核、步长和填充,计算 2D 卷积的空间输出尺寸、感受野及每像素 FLOPs。
如何使用 CNN 卷积输出尺寸计算器
- 输入空间尺寸。
- 输入卷积核大小、步长和填充。
- 点击计算,查看输出尺寸和感受野。
使用场景
- •为嵌入式设备手动设计 CNN 尺寸。
- •调试 PyTorch 中的形状不匹配问题。
- •讲解卷积基础原理。
公式
输出 = ⌊(输入 + 2 × 填充 − 卷积核) / 步长⌋ + 1。单层感受野等于卷积核尺寸。
常见问题
什么是 same 填充?
使输出尺寸与输入尺寸相同的填充方式。对于卷积核大小 k 且步长为 1,所需填充量为 ⌊k/2⌋。
为什么输出必须是整数?
输出为小数表明填充设置不匹配。大多数框架会对结果向下取整,从而丢失最右/最下一行像素,这也是有时使用非对称填充的原因。
这包含偏置和通道吗?
不包含。将参数量和 FLOPs 乘以输入通道数与输出通道数的乘积可得到该层的总量,并为每个输出通道额外加上一个偏置项。