如何使用模板函数从缓冲区（T* 数据数组）创建 cv::Mat？

Posted 2023-02-21

【中文标题】如何使用模板函数从缓冲区（T* 数据数组）创建 cv::Mat？

【英文标题】：How to create cv::Mat from buffer (array of T* data) using a template function?

【发布时间】：2018-01-11 13:06:14

【问题描述】：

我想编写一个模板函数来将指针T* image 引用的数据复制到cv::Mat。我很困惑如何概括 T 和 cv_type 匹配。

template<typename T>
cv::Mat convert_mat(T *image, int rows, int cols) 
    // Here we need to match T to cv_types like CV_32F, CV_8U and etc.
    // The key point is how to connect these two
    cv::Mat mat(rows, cols, cv_types, image);
    return mat;

我是模板编程的新手，我很困惑如何实现T-cv_types对应。

有人知道吗？谢谢！！！

【问题讨论】：

我不太明白这个问题 - 当您不使用模板时，您使用什么逻辑将类型映射到“cv_types”？

你可以试试cv::Mat_<T>，它不需要cv_type，因为它是模板化的

【参考方案1】：

使用cv::DataType<T>::type。

这是一个例子。

// Create Mat from buffer 
#include <opencv2/opencv.hpp>
using namespace std;
using namespace cv;


/*
//! First version
//! 2018.01.11 21:16:32 (+0800)
template <typename T>
Mat createMat(T* data, int rows, int cols) 
    // Create Mat from buffer
    Mat mat(rows, cols, cv::DataType<T>::type);
    memcpy(mat.data, data, rows*cols * sizeof(T));
    return mat;

*/

//! Second version 
//! 2018.09.03 16:00:01 (+0800) 
template <typename T>
cv::Mat createMat(T* data, int rows, int cols, int chs = 1) 
    // Create Mat from buffer 
    cv::Mat mat(rows, cols, CV_MAKETYPE(cv::DataType<T>::type, chs));
    memcpy(mat.data, data, rows*cols*chs * sizeof(T));
    return mat;


int main()
    int    arr1[4] = 1,2,3,4;
    double arr2[4] = 1.1,2.2,3.3,4.4;

    Mat mat1 = createMat<int>(arr1, 2,2);
    Mat mat2 = createMat<double>(arr2, 2,2);
    cout << "Mat1:\n"<< mat1 <<endl;
    cout << "Mat2:\n"<< mat2 <<endl;

结果：

Mat1:
[1, 2;
 3, 4]
Mat2:
[1.1, 2.2;
 3.3, 4.4]

【讨论】：

不错的答案。我已经为这个添加了书签。我一直想要一种简化的方式来做到这一点。谢谢！