对接时同事反映,他的数据是直接从内存读的,只好把图像数据转成数组给他。
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是单通道)。
记录一下~