c++ - 如何将 vector 转换为 vector？

Question

假设我有一个 vector<Mat>称为 regionFeaMat和 regionFeaMat.size() == 81 .换句话说，regionFeaMat有81个大小相等的矩阵，regionFeaMat[0].rows==256和 regionFeaMat[0].cols==1 .我想转换 reginFeaMat至 vector<float> reginFeaVec .我尝试使用以下代码，但得到了错误的结果:

vector<float> regionFeaVec;
regionFeaVec.assign((float*)regionFeaMat[0].datastart, (float*)regionFeaMat[80].dataend);

Answer 1

您似乎做出了一些错误的假设。

std::vector确实将其元素连续存储在内存中，但是 cv::Mat是一个包含指向其内部缓冲区的指针的 header ，因此只有 vector<Mat> 中的指针是连续存储的，而不是 Mat 数据本身。因此，位于 (float*)regionFeaMat[0].dataend 之间的内存和 (float*)regionFeaMat[80].datastart是一些随机垃圾 - 如果它确实部分包含其他 Mat 的数据，那纯属运气。

由于上述原因，您不能将单行分配 vector 分配给任何其他 vector ，而必须分别插入每个垫子。尝试这样的事情:

// prevent vector reallocation after each Mat insertion:
regionFeaVec.reserve(regionFeaMat.size()*regionFeaMat[0].cols*regionFeaMat[0].rows);
for (int i = 0; i < regionFeaMat.size(); ++i)
{
    if (regionFeaMat[i].isContinuous())
        regionFeaVec.insert(
                           regionFeaVec.end(), 
                           (float*)regionFeaMat[i].datastart, 
                           (float*)regionFeaMat[i].dataend
                           );
    else
    {
        for (int j = 0; j < regionFeaMat[i].rows; ++j)
        {
                const float* row = regionFeaMat[i].ptr<float>(j);
                regionFeaVec.insert(regionFeaVec.end(), row, row + regionFeaMat[i].cols);
        }
    }
}

请注意，我正在检查特定 Mat 对象是否连续，因为根据 OpenCV docs ，在某些情况下，每一行的末尾可能包含间隙，在这种情况下，我们必须逐行读取矩阵。

这段代码可以简化，因为如果矩阵是连续的，我们可以按照上面引用的文档将其视为一维 vector :

// prevent vector reallocation after each Mat insertion:
regionFeaVec.reserve(regionFeaMat.size()*regionFeaMat[0].cols*regionFeaMat[0].rows);
for (int i = 0; i < regionFeaMat.size(); ++i)
{
    cv::Size size = regionFeaMat[i].size();
    if (regionFeaMat[i].isContinuous())
    {
        size.width *= size.height;
        size.height = 1;
    }
    for (int j = 0; j < size.height; ++j)
    {
        const float* row = regionFeaMat[i].ptr<float>(j);
        regionFeaVec.insert(regionFeaVec.end(), row, row + size.width);
    }
}

如果你想在更一般的情况下防止 vector 重新分配，你还必须改变计算传递给reserve()的元素数量的方法。 .我使用的方法假设所有 Mat 对象只有两个维度，所有对象都相等，因为这是您描述问题的方式。

此外，如果您想将其分配给 vector<float> , 请确保 regionFeaMat 的元素类型是CV_32F .