在大图像尺寸上测试FCN

Question

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

Answer 1

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

在内存中调整网络大小非常昂贵，尤其是在调整所有图层时，使用输入尺寸进行缩放。在你给出的情况下，这将使你的内存需求增加大约8倍。

检查您可以拥有的尺寸，在初始化网络时观察内存数据;我相信Caffe会逐层报告内存需求。