> 文章列表 > 卷积输出大小的计算公式

卷积输出大小的计算公式

卷积输出大小的计算公式

定义

mmm: feature map的大小
kkk: kernel_size大小
sss: stride大小
ppp: padding大小

公式

m′=⌊m−k+2×ps+1⌋m'=\\lfloor \\frac{m-k+2\\times p}{s}+1 \\rfloorm′=⌊sm−k+2×p​+1⌋

例如,如果是kenel_size=3=3=3的卷积,要想feature_map的尺寸减半,那么当stride=2=2=2时,padding是多少呢?则需要令
⌊m−3+2×p2+1⌋=m2\\lfloor \\frac{m-3+2\\times p}{2}+1 \\rfloor=\\frac{m}{2}⌊2m−3+2×p​+1⌋=2m​,又因为有

⌊m−3+2×p2+1⌋=⌊m−3+2×p+22⌋=⌊m−1+2×p2⌋\\lfloor \\frac{m-3+2\\times p}{2}+1 \\rfloor=\\lfloor \\frac{m-3+2\\times p+2}{2}\\rfloor=\\lfloor \\frac{m-1+2\\times p}{2}\\rfloor⌊2m−3+2×p​+1⌋=⌊2m−3+2×p+2​⌋=⌊2m−1+2×p​⌋

即:在m2\\frac{m}{2}2m​为整数的情况下,需要令⌊2×p−12⌋\\lfloor \\frac{2\\times p-1}{2}\\rfloor⌊22×p−1​⌋的值为0,=>=>=>p=1