在OpenCV中扩展数组

时间:2015-08-24 13:08:05

标签: opencv point-cloud-library

我想制作一个包含X,Y,Z,RGBA和Label字段的PCD文件。现在我有一个XYZRGBA,PCD文件。它包括640 * 480点。另一方面,我有另一个文件,其中包括320 * 256个数字,代表分段图像中的标签。我想升级标签数组并将其添加到我当前的PCD文件中,以生成具有相应x,y,z,rgba和标签信息的新PCD文件。此PCD文件将与分段图像相关。  这是我的尝试。  Label是包含标签信息的文件名,首先我将其转换为OpenCV矩阵,现在我想将其扩展到640 * 480,然后将其添加到当前的xyzrgba,pcd文件中。再次向上扩展后,我将得到的OpenCV矩阵转换为名称为" array"用于添加到我当前的PCD数据。

cv::Mat LabelsMat=cv::Mat(320,256, CV_32SC1, label); 
cv::Mat OriginalLabels=cv::Mat::zeros(320,256, CV_32SC1);
LabelsMat.copyTo(OriginalLabels);    
cv::Mat UpScaledLabels=cv::Mat::zeros(640, 480, CV_32FC1); 
cv::resize(OriginalLabels, UpScaledLabels, UpScaledLabels.size(), 0, 0,cv::INTER_NEAREST); 
std::vector<int> array;
array.assign((int*)UpScaledLabels.datastart, (int*)UpScaledLabels.dataend);

但是有一些问题。当我制作这个新的PCD文件并且只想看到图像的一个片段时,例如4,根据我的基本图像,出现了一个错误的形状,与段4非常不同。我确定问题是因为这部分和上面的代码。有人能帮我找到问题吗?感谢您的宝贵帮助。

1 个答案:

答案 0 :(得分:1)

好的,终于有时间......

查看您生成的Mat对象总是好的,只需使用cv :: imshow或cv :: imwrite并相应地缩放数据。

使用此代码(基本上是您自己的代码与固定数组写入):

int label[320][256];

std::ifstream in("../inputData/UPSCALE_data.dat"); 
for (int i = 0; i < 320; i++) 
    for (int j = 0;j< 256; j++) 
    { 
        in >> label[i][j]; 
    } 
in.close();

// create Mat with label input:
cv::Mat LabelsMat=cv::Mat(320,256, CV_32SC1, label); 

cv::Mat OriginalLabels = LabelsMat.clone(); // you could instead work on the non-copied data, if you liked to...

// upscale:
cv::Mat UpScaledLabels; // no need to allocate memory here during testing
cv::resize(OriginalLabels, UpScaledLabels, cv::Size(640, 480), 0, 0,cv::INTER_NEAREST); 

std::vector<int> marray;
marray.reserve(UpScaledLabels.cols*UpScaledLabels.rows);
for(int j=0; j<UpScaledLabels.rows; ++j)
    for(int i=0; i<UpScaledLabels.cols; ++i)
    {
        marray.push_back(UpScaledLabels.at<int>(j,i));
    }

// now here marray has information about the upscaled image.


cv::Mat convertedCorrect;
UpScaledLabels.convertTo(convertedCorrect, CV_8UC1);

cv::imwrite("../outputData/UPSCALED_RESULT_ORIG.png", convertedCorrect*50);

我得到了这个结果:

enter image description here

那是因为cv::Mat LabelsMat=cv::Mat(320,256, CV_32SC1, label);生成了一个高度为320且宽度为256的图像(我以为我已经在评论中提到过但是找不到它...)

使用此代码解决问题:

int label[320][256];

std::ifstream in("../inputData/UPSCALE_data.dat"); 
for (int i = 0; i < 320; i++) 
    for (int j = 0;j< 256; j++) 
    { 
        in >> label[i][j]; 
    } 
in.close();

// create Mat with label input: HERE THE DIMENSIONS ARE SWAPPED
cv::Mat LabelsMat=cv::Mat(256,320, CV_32SC1, label); 

cv::Mat OriginalLabels = LabelsMat.clone(); // you could instead work on the non-copied data, if you liked to...

// upscale:
cv::Mat UpScaledLabels; // no need to allocate memory here during testing
cv::resize(OriginalLabels, UpScaledLabels, cv::Size(640, 480), 0, 0,cv::INTER_NEAREST); 

std::vector<int> marray;
marray.reserve(UpScaledLabels.cols*UpScaledLabels.rows);
for(int j=0; j<UpScaledLabels.rows; ++j)
    for(int i=0; i<UpScaledLabels.cols; ++i)
    {
        marray.push_back(UpScaledLabels.at<int>(j,i));
    }

// now here marray has information about the upscaled image.


cv::Mat convertedCorrect;
UpScaledLabels.convertTo(convertedCorrect, CV_8UC1);

cv::imwrite("../outputData/UPSCALED_RESULT_CORRECTED.png", convertedCorrect*50);

你得到的Mat看起来好多了:

enter image description here

但与其他图像相比,该图像以某种方式旋转了?!?

enter image description here