你可以一次传递给Caffe多少张图片？

Question

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

Answer 1

唯一的限制是你机器的内存：当caffe加载模型时，它为所有参数和分配所有中间数据blob的内存。您同时处理的图像越多，您需要提前分配的内存越大。
确定这个数字的最简单（也是最粗鲁）的方法就是跟踪错误，尝试将其设置为200并查看是否得到“内存不足”＃34;加载模型时出错 请注意，您可以同时处理的图像数量还取决于您使用的是GPU还是CPU：通常GPU内存小于CPU内存，因此可以处理更少的图像。