利用opencv将raw转换为rgb

        最近工作中碰到将.raw格式文件转换为彩图格式文件。待转换的.raw格式文件不带有图像的宽高等信息，只有图像像素值信息，利用opencv接口无法直接读取。网上查的方法都无法满足我的需求，经过各种尝试终于成功了。因此，将该格式下的raw文件的转换方法总结下，给自己留个笔记，也希望对opencv的初学者有所帮助。

        raw数据是大多数相机（单板sensor）的原始图像数据，与像.bmp，.jpg等格式的彩图相比，他的图像像素只有单个通道，即r，gr，gb和b中的任意一个（gr和gb可以理解为都是g通道）。因此，将raw数据转换为正常的彩图，就要补充图像像素点缺失的另外两个通道的值，对应的处理算法称为去马赛克(demosaic)或者插值(Bayer Color Filter Array, CFA)。Bayer是raw格式较常用的一种，如图1，网上有较详细的资料，这里不做深入说明。

图1

        庆幸的是，opencv中的接口cvCvtColor帮我们做了从raw到rgb的转换。接下来的问题是，只要利用别的方法正确读取raw数据即可。目前我碰到的raw数据文件，文件未存储图像格式相关信息（比如宽和高），像素值内容以二进制文件形式存储，每个像素值存储空间为2个字节。因此，首先将图像像素值内容从二进制文件读取，然后拷贝到对应的符合opencv接口cvCvtColor转换的数据结构，最后利用该接口就可以得到结果。以1080p分辨率(宽：1920；高：1080)名字为"d6500_1920x1080"的raw为例，其bayer格式是Gb-B-R-GR，转换代码如下：

int raw2rgb(void)
{
	char *rawFileName = "./6500_1920x1080.raw";
	FILE *fp = NULL;
	int ret = 0, width = 1920, height = 1080;

	/*为读取14bit raw数据分配空间*/
	unsigned short *pRawData = (unsigned short *)calloc(width*height, sizeof(unsigned short));

	if (NULL == pRawData)
	{
		printf("Fail to calloc buf\r\n");
		return -1;
	}
    
	if (NULL == (fp=fopen(rawFileName, "rb")))
	{
		printf("Fail to read %s.\r\n", rawFileName);
		return -2;
	}

        ret = fread(pRawData,sizeof(unsigned short)*width*height,1, fp);
	if (ret != 1)
	{
		printf("Fail to read raw data\r\n");
		return -3;
	}
	
	IplImage *pBayerData = cvCreateImage(cvSize(width,height), 16, 1);
	IplImage *pRgbDataInt16 = cvCreateImage(cvSize(width,height),16,3);
	IplImage *pRgbDataInt8 = cvCreateImage(cvSize(width,height),8,3);
	memcpy(pBayerData->imageData, (char *)pRawData, width*height*sizeof(unsigned short));
	cvCvtColor(pBayerData, pRgbDataInt16, CV_BayerRG2BGR);

	/*将14bit数据转换为8bit*/
	cvConvertScale(pRgbDataInt16, pRgbDataInt8, 0.015625, 0);

	cvNamedWindow("rgb", 1);
	cvShowImage("rgb", pRgbDataInt8);
	cvWaitKey(0);

	free(pRawData);
	fclose(fp);
	cvDestroyWindow("rgb");
	cvReleaseImage(&pBayerData);
	cvReleaseImage(&pRgbDataInt8);
	cvReleaseImage(&pRgbDataInt16);


	return 0;
}

但是有一点我不明白的地方是，我的raw格式Gb-B-R-GR，但是用cvCvtColor转换参数是CV_BayerRG2BGR才能转换对。