如何将“mean_file_proto”读入 OpenCV Mat 以从平均值中减去图像以在 Caffe 模型中使用？

Question

我使用的是 OpenCV 3.2。

我下载了一个包含平均文件 mean.binaryproto 的 Caffe 模型。我想从我的输入图像中减去平均值。 如何将mean_file_proto 读入 OpenCV Mat 以从均值中减去图像以在 Caffe 模型中使用？

编辑：

我可以只使用 OpenCV 而不使用 Caffe 吗？我在 Windows 10 64 位的 Visual Studio 2015 中使用 OpenCV 3.2。我从网络摄像头得到一张图像，然后我想减去平均值，然后使用OpenCV::dnn 得到结果。

Answer 1

平均文件是平面的，可能是 1 或 3 个通道。比例为 [0, 255]。

caffe::BlobProto mean_proto;
caffe::ReadProtoFromBinaryFile( filename, &mean_proto ); // check return code
caffe::Blob< float > mean_blob;
mean_blob.FromProto( mean_proto );
float *data = mean_blob.mutable_cpu_data();

您现在可以为每个频道创建CV_32FC1 类型的cv::Mat。

cv::Mat channel( mean_blob.height(), mean_blob.width(), CV_32FC1, data );
data += mean_blob.height() * mean_blob.width();

最后，用这些通道填充一个向量，并使用cv::merge() 将它们组合成一个cv::Mat，您可以从包含输入图像的另一个cv::Mat 中减去它。如果您的图像已经标准化 [0, 1)，您还需要重新调整均值。