OpenCV 3D 垫到矢量

我有 3D Mat，想将其转换为 Vector。我尝试了 opencv 的 reshape() 函数，但它似乎不适用于维度超过 2 的矩阵。如何将 3D Mat 转换为 Vector ？我可以通过访问 Mat 中的所有元素来实现。有什么有效的方法吗？

最佳答案

如果我们有以下 3D 矩阵:

const int ROWS=2, COLS=3, PLANES=4;
int dims[3] = {ROWS, COLS, PLANES};
cv::Mat m = cv::Mat(3, dims, CV_32SC1);   // works with other types (e.g. float, double,...)

这仅适用于连续垫对象(即 m.isContinuous() == true)

要获取包含相同元素的相同类型的向量:

使用重载的 STL 向量构造函数:

声明和初始化与矩阵相同类型的向量并复制所有 mat 元素:

const int* p3 = m3.ptr<int>(0);
std::vector<int> flat0(p3, p3+m3.total());

使用 OpenCV 的重塑:

这与第一个解决方案的不同之处在于元素最终的布局方式。
最初来自这篇文章:How to divide 3D matrix into bunches of 2D matrix with opencv C++
如果可能的话，我支持他关于将维度定义为 channel 的建议。 OpenCV 比维度更好地处理 channel 。当然，如果您打算拥有超过 4 个维度，这可能不适用。

cv::Mat m2(ROWS, COLS*PLANES, CV_32SC1, m3.data); // no copying happening here
cv::Mat m2xPlanes = m2.reshape(PLANES); // not sure if this involves a copy
std::vector<Mat> planes;
cv::split(m2xPlanes, planes); // usually used for splitting multi-channel matrices

std::vector<int> flat;
for(size_t i=0; i<planes.size(); i++) {
    cv::Mat plane_i = planes[i];
    const int* plane_i_ptr = plane_i.ptr<int>(0);
    flat.insert(flat.end(), plane_i_ptr, plane_i_ptr+plane_i.total());
}

对于这两种解决方案，所有元素都被考虑在内，只是它们的排序方式不同，因此访问方式也不同。
首先，您可以通过以下方式访问行、列、平面上的元素

int index = row * COLS * PLANES + col * PLANES + p

在第二个中，您的元素按平面排序。

选择哪种解决方案可能取决于您将如何索引向量。

关于OpenCV 3D 垫到矢量，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/27702231/