卷积神经网络中特征图大小计算公式总结

W:输入特征图的宽,H:输入特征图的高

K:kernel size卷积核宽和高,P:padding(特征图需要填充的0的个数),S:stride步长

width_out:卷积后输出特征图的宽,height_out:卷积后输出特征图的高

普通卷积

计算公式:

width_out = (W - K + 2 * P)/ S + 1(向下取整)

height_out = (H - K + 2 * P) / S + 1(向下取整)

池化

计算公式:

width_out = (W - K)/ S + 1(向下取整)

height_out = (H - K) / S + 1(向下取整)

上采样UpSampling2D

上采样相当于放大多少倍,size=倍数

计算公式:

width_out = W * size

height_out = H * size

转置卷积

转置卷积俗称反卷积,是上采样方式中的一种,转置卷积用来增大特征图的分辨率。

计算公式:

width_out = (W - 1)* S - 2 * P + K

height_out = (H - 1)* S - 2 * P + K

猜你喜欢

转载自blog.csdn.net/L888666Q/article/details/128705452