C语言中BYTE*指针类型转OpenCV中Mat类型

把byte类型转成OpenCV mat类型进行图像运算,或者把mat类型转为byte类型进行图片传输时，下面给出两者互相转换的函数

//Byte *转Mat
//inputParam:pImg,height,wide,channel
//outputParam:output_image
bool ByteToMat(BYTE* pImg, int height, int wide, int channel, cv::Mat& output_img)
{
	if (pImg == nullptr)
	{
		return false;
	}
	int nByte = nH * nW * nChannel / 8;
	int type = channel == 1 ? CV_8UC1 : CV_8UC3;
	output_img = cv::Mat::zeros(height, wide, type);
	memcpy(output_img.data, pImg, nByte);
	return true;
}

//Mat转Byte*
//inputParam:input_img
//outputParam:pImg,height,wide,channel
bool MatToByte(cv::Mat& input_img, BYTE*& pImg,int& wide,int& height,int& channel)
{
	channel = input_img.channels() ;
	height = input_img.rows;
	width = input_img.cols;
	if (pImg)
		delete[] pImg;
	pImg = new BYTE[height * width * channel];	
	pImg=input_img.data;
	return true;
}