一次可以向 Caffe 传递多少图像？

Question

我注意到 Caffe MNIST example prototxt file 如何允许一次最多将 64 张图像传递到网络。
我可以将这个数字设置多高有限制吗？
我可以（例如）将此数字设置为 200 甚至 500，以便我可以一次接受最多 200/500 张图像而不会对预测产生负面影响吗？

Answer 1

唯一的限制是您机器的内存：当 caffe 加载模型时，它会为所有参数 和 所有中间数据 blob 分配内存。并发处理的图片越多，需要提前分配的内存就越大。
确定此数字的最简单（也是最粗略）的方法就是试错法，尝试将其设置为 200 并查看加载模型时是否出现 "out of memory" 错误。
请注意，您可以同时处理的图像数量还取决于您使用的是 GPU 还是 CPU：通常 GPU 内存小于 CPU 内存，因此可以处理更少的图像。