为 Caffe 缩放 OpenCV Mat

Scaling OpenCV Mat for Caffe

我一直在关注 Caffe MINST example 并尝试使用 C++ 部署训练模型的测试,我使用 OpenCV 读取图像。在这个例子中,他们提到了他们

如何训练和测试图像

scale the incoming pixels so that they are in the range [0,1). Why 0.00390625? It is 1 divided by 256.

我听说 Caffe 中有一个 DataTransformer class 可以用来缩放图像,但是如果我将 OpenCV Mat 对象中的每个像素乘以 0.00390625 会得到相同的结果吗?

这个想法是对的。但请记住在缩放之前将您的 OpenCV Mats 转换为 float 或 double 类型。

类似于:

cv::Mat mat; // assume this is one of your images (grayscale)

/* convert it to float */
mat.convertTo(mat, CV_32FC1); // use CV_32FC3 for color images

/* scaling here */
mat = mat * 0.00390625;

更新 #1:转换和缩放也可以简单地在一行中完成,即

cv::Mat mat; // assume this is one of your images (grayscale)

/* convert and scale here */
mat.convertTo(mat, CV_32FC1, 0.00390625);