machine-learning - 一次可以向 Caffe 传递多少张图像?

标签 machine-learning computer-vision neural-network deep-learning caffe

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

最佳答案

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

关于machine-learning - 一次可以向 Caffe 传递多少张图像?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/32589349/

相关文章:

python - 不知道如何将 sklearn 与包含文本和数字的特征向量一起使用

machine-learning - 如何在arff文件中表示n-gram特征?

opencv - 数手指数

opencv - 使用SVD时,如何对Sigma对角线的比例因子进行分类?

python - 训练期间验证准确率突然下降

machine-learning - 神经网络的批量大小是多少?

从头开始实现的 C++ 神经网络在 MNIST 上无法达到 50% 以上

python-2.7 - 使用 k 均值进行图像分割

artificial-intelligence - 选择正确的神经网络类型

matlab - 粒子群优化训练神经网络的概念问题