使用 pytorch 预训练模型进行预测时所需的图像大小

Question

我使用 PyTorch 的 ResNet-18 来预测图像。我读到 (224, 224) 是该模型的图像尺寸。但是当我尝试将图像调整为 (124, 124) 或 (324, 324) 时，它仍然有效。谁能告诉我为什么？

Answer 1

PyTorch 上 ResNet 变体的实现在全连接层之前带有一个 AdaptiveAvgPool2d 层，确保无论输入大小如何，输出特征对于全连接层始终具有正确的形状。

此外，建议输入大小 224x224 以防止填充量不理想。