如何确定CNN的input尺寸?
可以先用torchsummary.summary(net,input_data=[B,H,W])清晰的观察网络的结构以及特征图尺寸变化。此处的[B,H,W]可以任意指定一个。例如[1,256,512]。然后根据打印出来的信息,用“H或W” / “最小的特征尺寸”,例如256/32=8=2^3(说明该网络有3个池化层),可以得知该网络的输入图像尺寸应该crop为8的倍数。如果已知pool的个数是n,则input尺寸应该是2^n的整数倍。input尺寸与pool池化层个数有关。
原创
2024-05-14 11:12:31 ·
167 阅读 ·
0 评论