caffe - 在大图像尺寸上测试 FCN

Question

我正在 Caffe 中测试 FCN-VGG16 网络。我没有将输入图像的大小调整为固定大小，而是每次都将网络重塑为图像大小。我在 matcaffe 和 pycaffe 中都试过这个。在这两种情况下，它似乎都能够运行小图像（例如，500x500）。但是，当我有更大尺寸的图像（例如 1920 x 1080）时，我会收到此错误“检查失败：错误 == cudaSuccess (2 vs. 0) out of memory”。我在具有 12 GB 内存的 Quadro M6000 GPU 上运行它。任何帮助/建议将不胜感激。

Answer 1

我知道从消息中可以看出这一点，但是您确实必须减小输入大小：GPU 没有足够的板载内存来存储您输入的所有内容。你能减少你的批量吗？如果没有，你真的需要整个 2M 像素吗？大多数模型在裁剪或缩小的图像方面做得很好。作为输入处理的一部分（在 CPU 上）执行此操作，并将其参数化为任何正弦输入。

调整网络大小在内存中是昂贵的，特别是如果您调整所有层的大小，并随着输入尺寸进行缩放。在您提供的情况下，这将使您的内存需求增加大约 8 倍。

作为检查你可以拥有的大小，在初始化网络时观察内存数据；我相信 Caffe 按层报告内存需求。