Opencv Mat与数组转换 C++

对接时同事反映,他的数据是直接从内存读的,只好把图像数据转成数组给他。

uint8_t* Mat_to_array(const cv::Mat input)
{
	int height = input.rows;
	int width = input.cols;
	uint8_t* pRgb = new uint8_t[height * width * 3];
	for (int i = 0;i < height;i++)
	{
		for (int j = 0;j < width;j++)
		{
			for (int k = 0;k < 3;k++)
			{
				pRgb[i * width * 3 + j * 3 + k] = input.at<cv::Vec3b>(i, j)[k];
			}
		}
	}
	return pRgb;
}

后面他又叫我把后处理的结果画出来,我只好把数组转成Mat再进行下一步操作。

	/* 
	Array -> Mat 
	输入:
	uint8_t pRgb[],Rgb图片数据
	uint32_t width,表示Rgb图片对应宽度
	uint32_t height,表示Rgb图片对应高度
	uint32_t sizeYuv,表示pRgb数组大小
	*/
	Mat array_to_Mat(uint8_t* pRgb, uint32_t width, uint32_t height, uint32_t sizeRgb)
	{
		Mat rgbImg;
		rgbImg.create(height, width, CV_8UC3);
		memcpy(rgbImg.data, pRgb, sizeRgb);
		return rgbImg;
	}

这里sizeRgb = width * height * 3, 还有CV_8UC3这个参数要注意,是三通道。

如果是yuv数据的话就是CV_8UC1.而且height 要乘上3/2.

	cv::Mat yuvImg;
	yuvImg.create(height* 3 / 2, width, CV_8UC1);
	memcpy(yuvImg.data, pYuv, sizeYuv);

sizeYuv = yuvWidth * yuvHeight = rgbWidth * rgbHeight * 3/2(Yuv是单通道)。

记录一下~

你可能感兴趣的:(深度学习)