【问题标题】：How many images can you pass to Caffe at a time?一次可以向 Caffe 传递多少张图片？
【发布时间】：2015-12-11 21:41:28
【问题描述】：
我注意到 Caffe MNIST example prototxt file 一次最多允许将 64 张图像传递到网络。
我可以将这个数字设置到多高有限制吗？
我是否可以（例如）将此数字设置为 200 甚至 500，以便我一次最多可以接受 200/500 张图像而不会对预测产生负面影响？
【问题讨论】：
标签：
machine-learning
computer-vision
neural-network
deep-learning
caffe
【解决方案1】：
唯一的限制是您机器的内存：当 caffe 加载模型时，它会为所有参数和所有中间数据块分配内存。您同时处理的图像越多，您需要提前分配的内存就越大。
确定此数字的最简单（也是最粗略）的方法就是跟踪错误，尝试将其设置为 200，看看在加载模型时是否出现“内存不足”错误。
请注意，您可以同时处理的图像数量还取决于您使用的是 GPU 还是 CPU：通常 GPU 内存小于 CPU 内存，因此您可以处理更少的图像。