nn.conv2d(256, 256, 3, 3)
假设输入大小为[3, 128, 128]
,输出的大小则为128-3+1=126
,这一层参数个数为256*3*3+256=2560
.
假设输入尺寸为[C_in, H, W],则输出大小计算公式为[C_out, (H-K+2*P)/S+1, (W-K+2*P)/S+1]
。其中C_in
是输入通道数,H,W
是数据的尺寸,K
是核大小,P
是padding大小,S
是步长stride大小。
nn.conv2d(C_in, C_out, K, K)
参数个数的计算公式为C_out*K*K+C_out
。K
是核大小,pytorch中默认多少个输出就有多少个偏置。