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

卷积输出大小的计算公式

卷积输出大小的计算公式

定义

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=smk+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}2m3+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}\\rfloor2m3+2×p+1=2m3+2×p+2=2m1+2×p

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