Caffe中的批處理模式

Question

我想使用Caffe庫提取圖像特征，但是出現性能問題。 我只能使用CPU模式。 有人告訴我Caffe支持的批處理模式，在該模式下，處理一張圖像所需的平均時間要慢得多。

我正在調用以下方法：

const vector<Blob<Dtype>*>& 
Net::Forward(const vector<Blob<Dtype>* > & bottom, Dtype* loss = NULL);

我要放入一個大小為1的向量，其中包含一個具有以下尺寸的Blob-（數字：10，通道：3，寬度：227，高度：227）。 它代表與官方python包裝器相同的方式對單個圖像進行過采樣。

這可以工作並給出正確的結果。 但是，它太慢了。

每當我嘗試發送包含多個（相同尺寸）斑點的矢量時，都會出現以下錯誤：

F0910 16：10：14.848492 15615 blob.cpp：355]試圖復制不同大小的blob。
檢查故障堆棧跟蹤：

如何使Caffe批量處理圖像？

Answer 1

如果要喂入更大的批次，則需要bottom的第一個（也是唯一的）blob的num>10 。 饋入num=20的blob與饋入oversample=10兩個輸入相同。 當然，您將必須根據所使用的oversampling手動執行平均。

此外，您可能希望將deploy.prototxt文件中的第一個輸入維從10更改為更大的值（取決於計算機的內存容量）