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

Question

我使用 PyTorch 的 ResNet-18 来预测图像。我读过(224, 224) 是这个模型的图像大小。但是当我尝试将图像大小调整为(124, 124) 或(324, 324) 时，它仍然可以工作。谁能告诉我为什么？

Answer 1

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

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