2014-09-23 96 views
0

将矢量分配给循环中的矩阵行的最快方法是什么?我想用矢量沿着它的行填充一个数据矩阵。这些向量是循环计算的。这个循环持续到数据矩阵的所有条目填充那些向量。Opencv Mat矢量分配到一个矩阵的行,最快的方式?

目前我正在使用cv::Mat::at<>()方法来访问矩阵的元素并填充矢量,但看起来这个过程很慢。我尝试了另一种方式,使用cv::Mat::X.row(index) = data_vector,它工作的很快,但用我无法理解的一些垃圾值填充我的矩阵X,为什么。

我读到,存在另一种使用指针的方式(最快的方式),但是我无法理解。有人可以解释如何使用它们或其他不同的方法?

这里是我的代码的一部分:

#define OFFSET 2 

cv::Mat im = cv::imread("001.png", CV_LOAD_IMAGE_GRAYSCALE); 
cv::Mat X = cv::Mat((im.rows - 2*OFFSET)*(im.cols - 2*OFFSET), 25, CV_64FC1); // Holds the training data. Data contains image patches 
cv::Mat patch = cv::Mat(5, 5, im.type()); // Holds a cropped image patch 
typedef cv::Vec<float, 25> Vec25f; 

int ind = 0; 
for (int row = 0; row < (im.rows - 2*OFFSET); row++){ 
    for (int col = 0; col < (im.cols - 2*OFFSET); col++){ 

    cv::Mat temp_patch = im(cv::Rect(col, row, 5, 5)); // crop an image patch (5x5) at each pixel 
    patch = temp_patch.clone(); // Needs to do this because temp_patch is not continuous in memory 
    patch.convertTo(patch, CV_64FC1); 

    Vec25f data_vector = patch.reshape(0, 1); // make it row vector (1X25). 
    for (int i = 0; i < 25; i++) 
    { 
     X.at<float>(ind, i) = data_vector[i]; // Currently I am using this way (quite slow). 
    } 

    //X_train.row(ind) = patch.reshape(0, 1); // Tried this but it assigns some garbage values to the data matrix! 
    ind += 1; 
    } 
} 
+0

如果您确实想要访问单个ROW,它始终是连续的afaik。 – Micka 2014-09-23 09:39:17

回答

0

只是在你的代码夫妇编辑的

double * xBuffer = X.ptr<double>(0); 
for (int row = 0; row < (im.rows - 2*OFFSET); row++){ 
    for (int col = 0; col < (im.cols - 2*OFFSET); col++){ 

    cv::Mat temp_patch = im(cv::Rect(col, row, 5, 5)); // crop an image patch (5x5) at each pixel 
    patch = temp_patch.clone(); // Needs to do this because temp_patch is not continuous in memory 
    patch.convertTo(patch, CV_64FC1); 
    memcpy(xBuffer, patch.data, 25*sizeof(double)); 
    xBuffer += 25; 
    } 
} 

此外,您好像做补丁的任何计算只是提取灰度值,所以您可以创建与im相同类型的X,并在最后将其转换为double。通过这种方式,你可以memcpy的你的补丁的每一行,在内存beeing`无符号字符*缓冲= im.ptr(行)+山坳

+1

有些事情要提到:patch.convertTo(patch,...)将始终在循环的每次运行中重新分配内存。在循环之前创建另一个Mat:'cv :: Mat convertedPatch;'并重新使用它,内存不会被重新分配。如果'temp_patch.convertTo(patch,...)'(或convertedPatch)被直接使用,'Clone'不是必须的 - 如果它仍然连续,那么不能100%确定。 – Micka 2014-09-23 09:37:56

1

要做到这一点的常规方式OpenCV的,你可以做地址: -

ImageMat.row(RowIndex) = RowMat.clone(); 

RowMat.copyTo(ImageMat.row(RowIndex)); 

没有测试的正确性和速度。