OpenCV 3D 垫到矢量

Question

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

Answer 1

如果我们有以下 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,...)

这仅适用于连续的垫子对象（iemisContinuous() == true）

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

1. 使用重载的 STL 向量构造函数：

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

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

2. 使用 OpenCV 的重塑：

   这与第一个解决方案的不同之处在于元素最终的布局方式。最初来自这篇文章：如何使用 opencv C++ 将 3D 矩阵划分为成束的 2D 矩阵， 如果可能的话，我提出了他关于将维度定义为通道的建议。OpenCV 比维度更好地处理通道。当然，如果您打算拥有超过 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

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

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