opencv读取自定义bmp灰度图片

项目中图片文件非常大,是很多张图片(灰度图)的数据都放在一个此文件中,其实文件的头部还是bmp头部。

用opencv里边的cvLoadImage的话只能读取第一张图片的数据,因为读取图片的数据的多少是由文件头部的宽(width)高(height)决定的。

于是就想能不能fopen该文件然后直接定位到文件的数据部分,然后把该部分的数据copy到opencv的imageData中,就能使用opencv显示了。比如想读第二张图片的数据,就跳到第一张图片数据的末尾处就可以按照图片size的大小读数据了(注意读bmp数据的时候要确保宽width被4整除)。原始灰度图像:

opencv读取自定义bmp灰度图片_第1张图片


初始代码:

static BITMAPFILEHEADER fileHeader;
static BITMAPINFOHEADER infoHeader;
void ReadBmp(const string& bmpPath)
{
	/* 读取文件 */
	FILE * fp = fopen(bmpPath.c_str(), "rb");
	assert(fp != NULL);

	/* 跳过文件头 */
	fseek(fp, sizeof(BITMAPFILEHEADER),0);

	/* 读取信息头 */
	fread(&infoHeader, sizeof(BITMAPINFOHEADER), 1,fp);

	int bmpWidth = infoHeader.biWidth;
	int bmpHeight = infoHeader.biHeight;
	int biBitCount = infoHeader.biBitCount;				
	int lineByte=(bmpWidth * biBitCount/8+3)/4*4;		/* 定义变量,计算图像每行像素所占的字节数(必须是4的倍数)*/
	//申请位图数据所需要的空间,读位图数据进内存
	unsigned char* pBmpBuf=new unsigned char[lineByte * bmpHeight];

	fread(pBmpBuf,1,lineByte * bmpHeight,fp);		/* 可以在这里定位fp指针来读取第N张图片,这里是第一张,没有用fseek定位 */

	fclose(fp);//关闭文件


	IplImage* pImg = cvCreateImage(cvSize(bmpWidth, bmpHeight), 8, 1);
	
	assert(pImg != NULL);

	/* 复制图像数据到opencv数据结构总并显示 */
	memcpy(pImg->imageData, pBmpBuf, lineByte * bmpHeight * 
		sizeof(unsigned char));
	/* 直接拷贝图片数据到OpenCV结构里的话需要翻转 */
	cvFlip(pImg, NULL, 0);

	cvShowImage("pImg", pImg);
	cvWaitKey(0);
	free(pBmpBuf);
}

代码执行后的图像(很尴尬):

opencv读取自定义bmp灰度图片_第2张图片


可以看出图片完全不对劲,图片底部有奇怪的花纹,而且飞机轮廓也不在正中间了,应该是飞机右边的空白被挪到了左边。

如果把图片换成彩色图,不会出现这种情况。想到灰度图,或者深度小于24的图,会有颜色表,于是就想到了是不是因为颜色表没有读取,颜色表位于头部之后,图片数据之前。读取颜色表后的函数代码如下:

static BITMAPFILEHEADER fileHeader;
static BITMAPINFOHEADER infoHeader;
void ReadBmp(const string& bmpPath)
{
	/* 读取文件 */
	FILE * fp = fopen(bmpPath.c_str(), "rb");
	assert(fp != NULL);

	/* 跳过文件头 */
	fseek(fp, sizeof(BITMAPFILEHEADER),0);

	/* 读取信息头 */
	fread(&infoHeader, sizeof(BITMAPINFOHEADER), 1,fp);

	int bmpWidth = infoHeader.biWidth;
	int bmpHeight = infoHeader.biHeight;
	int biBitCount = infoHeader.biBitCount;				
	int lineByte=(bmpWidth * biBitCount/8+3)/4*4;		/* 定义变量,计算图像每行像素所占的字节数(必须是4的倍数)*/

	/* 灰度图像有颜色表,且颜色表表项为256,关键 */
	RGBQUAD* pColorTable;
	if(biBitCount==8)
	{

		//申请颜色表所需要的空间,读颜色表进内存

		pColorTable=new RGBQUAD[256];

		fread(pColorTable,sizeof(RGBQUAD),256,fp);

	}
	//申请位图数据所需要的空间,读位图数据进内存
	unsigned char* pBmpBuf=new unsigned char[lineByte * bmpHeight];

	fread(pBmpBuf,1,lineByte * bmpHeight,fp);

	fclose(fp);//关闭文件


	IplImage* pImg = cvCreateImage(cvSize(bmpWidth, bmpHeight), 8, 1);
	
	assert(pImg != NULL);

	/* 复制图像数据到opencv数据结构总并显示 */
	memcpy(pImg->imageData, pBmpBuf, lineByte * bmpHeight * 
		sizeof(unsigned char));
	/* 直接拷贝图片数据到OpenCV结构里的话需要翻转 */
	cvFlip(pImg, NULL, 0);

	cvShowImage("pImg", pImg);
	cvWaitKey(0);
	free(pBmpBuf);
}

代码运行后的图片:

opencv读取自定义bmp灰度图片_第3张图片


一切正常。

所以数据bmp的结构还是关键,特别是颜色表,图片的位色小于24时,都会有颜色表(跟我们本来想的不一样,不是彩色图片才有颜色表)。


有关bmp头部的介绍,以下这篇文章http://blog.csdn.net/lyy289065406/article/details/6717679介绍得不错,里边还介绍了普通bmp图片的读取与存储,这里也简单贴以下savebmp()的代码:

//-----------------------------------------------------------------------------------------
//给定一个图像位图数据、宽、高、颜色表指针及每像素所占的位数等信息,将其写到指定文件中
bool saveBmp(char *bmpName, unsigned char *imgBuf, int width, int height, int biBitCount, RGBQUAD *pColorTable)
{

    //如果位图数据指针为0,则没有数据传入,函数返回

    if(!imgBuf)
        return 0;

    //颜色表大小,以字节为单位,灰度图像颜色表为1024字节,彩色图像颜色表大小为0

    int colorTablesize=0;

    if(biBitCount==8)
        colorTablesize=1024;

    //待存储图像数据每行字节数为4的倍数

    int lineByte=(width * biBitCount/8+3)/4*4;

    //以二进制写的方式打开文件

    FILE *fp=fopen(bmpName,"wb");

    if(fp==0)
		return 0;

    //申请位图文件头结构变量,填写文件头信息

    BITMAPFILEHEADER fileHead;

    fileHead.bfType = 0x4D42;//bmp类型

    //bfSize是图像文件4个组成部分之和

    fileHead.bfSize= sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) + colorTablesize + lineByte*height;

    fileHead.bfReserved1 = 0;

    fileHead.bfReserved2 = 0;

    //bfOffBits是图像文件前3个部分所需空间之和

    fileHead.bfOffBits=54+colorTablesize;

    //写文件头进文件

    fwrite(&fileHead, sizeof(BITMAPFILEHEADER),1, fp);

    //申请位图信息头结构变量,填写信息头信息

    BITMAPINFOHEADER head; 

    head.biBitCount=biBitCount;

    head.biClrImportant=0;

    head.biClrUsed=0;

    head.biCompression=0;

    head.biHeight=height;

    head.biPlanes=1;

    head.biSize=40;

    head.biSizeImage=lineByte*height;

    head.biWidth=width;

    head.biXPelsPerMeter=0;

    head.biYPelsPerMeter=0;

    //写位图信息头进内存

    fwrite(&head, sizeof(BITMAPINFOHEADER),1, fp);

    //如果灰度图像,有颜色表,写入文件 

    if(biBitCount==8)
        fwrite(pColorTable, sizeof(RGBQUAD),256, fp);

    //写位图数据进文件

    fwrite(imgBuf, height*lineByte, 1, fp);

    //关闭文件

    fclose(fp);

    return 1;

}

以上代码通过手动设置bmp头部信息,更简单的方法就是读取另外一张正常的bmp图像把正常的头部读取出来,深度和高宽自己修改一下即可。


你可能感兴趣的:(C/C++,OpenCV,bmp)