深度学习中的图像分辨率

Question

让我总结一下我的问题，

每当我尝试使用我的自定义数据集训练模型时，我都遇到内存错误。后来，我注意到与同一数据集中的其他图像相比，某些图像的分辨率非常高。然而它们的尺寸并没有那么大。

我的预训练模型中有图像缩放器，所以我认为我上面提到的情况不会有问题，但我不能确定。它会导致问题吗？

Answer 1

是的，它可能是内存错误的潜在来源。通常，内存错误的发生有两个原因，第一个是图像尺寸过大（甚至调整后的尺寸也很高）和批量大小。

这些在某种意义上是相关的。如果使用较大的图像尺寸，则必须使用较小的批处理尺寸来补偿内存。但是，不要将图像尺寸减小太多。如果您认为尽管图像大小合理但仍面临内存错误，则必须减少批处理大小。通常，输入图像大小因型号而异。