vector数组转opencv的mat

在itk中需要将单层图像转为opencv格式进行展示，又没有办法直接转，只能将数据存在vector数组中，然后转为opencv的mat格式

common.h

#pragma once

#include<vector>
#include<opencv2/opencv.hpp>

/// <summary>
/// vectot数组转opencv的mat
/// </summary>
/// <param name="vinput">输入vector</param>
/// <param name="channels">图像通道数</param>
/// <param name="rows">图像h方向大小</param>
/// <param name="mout">输出图像</param>
void vector2mat(
	const std::vector<double> vinput, 
	const int channels, const int rows,
	cv::Mat& mout)
{
	cv::Mat mat = cv::Mat(vinput);
	mout = mat.reshape(channels, rows).clone();
}

main.cpp

#include<iostream>
#include"common.h"
using namespace std;

int main()
{
	vector<double> gray_img_value;
	for (int i = 0; i < 400 * 400; i++) {
		double value = (rand() % 255) / 255.0;
		gray_img_value.push_back(value);
	}
	cv::Mat img;
	vector2mat(gray_img_value, 1, 400, img);

	cout << img.size << endl;
	cout << img.channels() << endl;

	cv::imshow("w", img);
	cv::waitKey();	

	return 0;
}