BMP2YUV实验报告

crema_h

已于 2022-06-24 01:03:20 修改

阅读量492

点赞数

文章标签：图像处理人工智能

于 2022-06-24 00:41:45 首次发布

本文链接：https://blog.csdn.net/weixin_46352852/article/details/125437252

版权

本文详细介绍了BMP文件格式，包括位图头文件和信息头结构，以及RGB到YUV色彩空间转换的原理。接着，提供了一段C++代码实现BMP文件到YUV文件的转换，包括读取BMP文件、RGB转YUV、YUV数据存储等步骤。最后，展示了实验结果和参考资料，用于验证和理解代码的正确性。

摘要由CSDN通过智能技术生成

一、实验要求

1.自行生成多个BMP文件，至少含5个不同场景画面，需要带含有班级、学号后四位和本人姓名的logo。

2.编写将第一步所生成的多个BMP文件转化为YUV文件，要求可在命令行中设置每个画面出现的帧数。最后形成的YUV文件应至少包含200帧。重点掌握函数定义、缓冲区分配、倒序读写、结构体的操作。

3.对整个程序进行调试，并将生成的YUV文件用播放软件观看，验证是否正确。

二、BMP格式

BMP（全称 Bitmap）是 Windows 操作系统中的标准图像文件格式，可以分成两类：设备相关位图（DDB）和设备无关位图（DIB）。它采用位映射存储格式，除了图像深度可选以外，在绝大多数应用中不采用其他任何压缩，因此BMP 文件占用空间很大。

BMP 文件图像深度可选 1bit、4bit、8bit、16bit 、24bit及32bit。BMP 文件存储数据时，图像的扫描方式是按从左到右、从下到上的顺序。

典型的 BMP 图像文件由四部分组成：

（1）位图头文件数据结构，它包含 BMP 图像文件的类型、显示内容等信息

typedef struct tagBITMAPFILEHEADER {
	WORD bfType; /* 说明文件的类型*/
	DWORD bfSize;/* 说明文件的大小，用字节为单位*/
	WORD bfReserved1; /* 保留，设置为0 */
	WORD bfReserved2; /* 保留，设置为0 */
	DWORD bfOffBits; /* 说明从BITMAPFILEHEADER结构开始到实际的图像数据之间的字节偏移量*/
	} BITMAPFILEHEADER;

（2）位图信息数据结构，它包含有 BMP 图像的宽、高、压缩方法，以及定义颜色等信息

typedef struct tagBITMAPINFOHEADER {
	DWORD biSize; /* 说明结构体所需字节数*/
	LONG biWidth; /* 以像素为单位说明图像的宽度*/
	LONG biHeight; /* 以像素为单位说明图像的高速*/
	WORD biPlanes; /* 说明位面数，必须为1 */
	WORD biBitCount; /* 说明位数/像素，1、2、4、8、24 */
	DWORD biCompression; /* 说明图像是否压缩及压缩类型BI_RGB，BI_RLE8，BI_RLE4，BI_BITFIELDS */
	DWORD biSizeImage; /* 以字节为单位说明图像大小，必须是4的整数倍*/
	LONG biXPelsPerMeter; /*目标设备的水平分辨率，像素/米*/
	LONG biYPelsPerMeter; /*目标设备的垂直分辨率，像素/米*/
	DWORD biClrUsed; /* 说明图像实际用到的颜色数，如果为0，则颜色数为2的biBitCount次方*/
	DWORD  biClrImportant;  /*说明对图像显示有重要影响的颜色索引的数目，如果是0，表示都重要。*/
	} BITMAPINFOHEADER;

（3）调色板(可选)，有些位图需要调色板，但真彩色图（24位BMP）不需要

typedef struct tagRGBQUAD {
	BYTE rgbBlue; /*指定蓝色分量*/
	BYTE rgbGreen; /*指定绿色分量*/
	BYTE rgbRed; /*指定红色分量*/
	BYTE rgbReserved; /*保留，指定为0*/
	} RGBQUAD;

（4）位图数据，这部分的内容根据 BMP 位图使用的位数不同而不同

对于用到调色板的位图，图像数据是该像素颜色在调色板中的索引值。

对于真彩色图，图像数据是实际的 R、G、B值（按BGR顺序依次排列）。图像的每一扫描行由表示图像像素的连续的字节组成，每一行的字节数取决于图像的颜色数目和用像素表示的图像宽度。规定每一扫描行的字节数必须是4的整倍数。扫描行是由底向上存储的，阵列中的第一个字节表示位图左下角的像素，而最后一个字节表示位图右上角的像素。

在BMP文件中，如果一个数据需要用几个字节来表示的话，那么该数据的存放字节顺序为“低地址村存放低位数据，高地址存放高位数据”。如数据0x1756在内存中的存储顺序如下图，这种存储方式称为小端方式(little endian) ，与之相反的是大端方式（big endian）。

三、RGB2YUV

RGB到色差信号的转换如下所示：

Y＝0.2990R+0.5870G+0.1140B

R-Y＝0.7010R-0.5870G-0.1140B

B-Y＝-0.2990R-0.5870G+0.8860B

为了使色差信号的动态范围控制在0.5之间，需要进行归一化，对色差信号引入压缩系数。归一化后的色差信号为：

U＝-0.1684R-0.3316G+0.5B

V＝0.5R-0.4187G-0.0813B

对亮度信号和色度信号进行8比特均匀量化时，需考虑码电平分配。对亮度信号，共分为256个等间隔量化级，其中0~15级和236~255级为信号保护带，16~235级为使用的亮度电平；对色差信号，归一化后动态范围为-0.5~0.5,256个等间隔量化级中，128级对应色差零电平，16~240级为色差信号电平，0~15级和241~255级为保护带。

将RGB信号转换为4:2:0格式的YUV文件时，色差信号U、V的取样频率为亮度信号取样频率的1/4，在垂直和水平方向的取样点数为亮度信号Y的一半。因此需要根据公式计算得到一帧Y、U、V的数据后，要对色差信号进行下采样（即相邻正方形四个值取平均）。

四、代码实现

实验流程：

（1）读位图文件头，判断是否是bmp文件，是否可读出

（2）读位图信息头，判断是否可读出，若biBitCount不为24，还应关注biSize，它和文件头一起指出了调色板实际开始的位置；

（3）开辟缓存区，读取bmp文件数据rgb，y，u，v的缓存区大小固定，分别为

biWidth×biHeighet×3，biWidth×biHeighet，biWidth×biHeighet/4，biWidtht×biHeighet/4

bmp缓存区大小应为biWidth×biHeighet×biBitCount/8，但也可根据读取的字节数另外设置，这里的biWidth,biHeight一定要是4的倍数，如果不是将其转换为4的倍数。

（4）根据bmp图像位数，选择相应操作，将数据倒序存放到rgb缓存区中

（5）调用RGB2YUV函数得到YUV文件。

4.1 main.cpp

#include<stdio.h>
#include"bmp2yuv.h"
#include<stdlib.h>
#include<windows.h>


int main(int argc, char* argv[])  //argc:命令行总参数个数
{
	BITMAPFILEHEADER file_header;  //定义两个结构体变量
	BITMAPINFOHEADER info_header;

	FILE* bmpFile = NULL;   
	FILE* yuvFile = NULL;

	int frameWidth;
	int frameHeight;
	int framenumber;
	int count;  //图像的重复帧数
	int sum = 0;  //合成的yuv总帧数
	unsigned short bitcount;  //位深
	unsigned char* bmpBuf, * rgbBuf, * yBuf, * uBuf, * vBuf;
	const int pic_num = 5;
	const char* bmpfile[pic_num] = { "1.bmp","2.bmp","3.bmp","4.bmp","5.bmp" };
	const char* yuvfile = "out.yuv";

	yuvFile = fopen(argv[11], "wb");  //打开输出文件，二进制写入yuv文件
	for (int i = 0; i < 2 * pic_num; i = i + 2)
	{
		bmpFile = fopen(argv[i + 1], "rb");
		printf("\n读取bmp文件:%s\n", argv[i + 1]);
		framenumber = atoi(argv[i + 2]);  //写入每个图像的重复帧数
		count = 0;

		//通过文件头判断是否为bmp文件
		if (fread(&file_header, sizeof(BITMAPFILEHEADER), 1, bmpFile) != 1)
		{
			printf("read file header error!\n");
			exit(0);
		}
		if (file_header.bfType != 0x4D42)
		{
			printf("not bmp file!\n");
			exit(0);
		}
		else
		{
			printf("this is a bmp file!\n");
		}

		//位图信息头
		if (fread(&info_header, sizeof(BITMAPINFOHEADER), 1, bmpFile) != 1)
		{
			printf("read info header error!\n");
			exit(0);
		}

		//读取图像宽、高、位数
		frameWidth = info_header.biWidth;
		frameHeight = info_header.biHeight;
		bitcount = info_header.biBitCount;

		//开辟缓冲区
		rgbBuf = (unsigned char*)malloc(frameWidth * frameHeight * 3);
		yBuf = (unsigned char*)malloc(frameWidth * frameHeight);
		uBuf = (unsigned char*)malloc((frameWidth * frameHeight) / 4);
		vBuf = (unsigned char*)malloc((frameWidth * frameHeight) / 4);

		//信息头中宽和高以像素为单位，转换为字节为单位
		int width = frameWidth * info_header.biBitCount / 8;
		int height = frameHeight;
		bmpBuf = ((unsigned char*)malloc(sizeof(unsigned char) * width * height));

		//文件指针读取文件有效数据
		fread(bmpBuf, width * height, 1, bmpFile);

		//读bmp文件得rgb数据
		bmp2rgb(bmpBuf, width, height, rgbBuf);

		//rgb转yuv
		rgb2yuv(frameWidth, frameHeight, rgbBuf, yBuf, uBuf, vBuf);

		//将重复次数的图像写入yuv文件
		for (int a = 0; a < framenumber; a++)
		{
			fwrite(yBuf, 1, frameWidth * frameHeight, yuvFile);
			fwrite(uBuf, 1, (frameWidth * frameHeight) / 4, yuvFile);
			fwrite(vBuf, 1, (frameWidth * frameHeight) / 4, yuvFile);

			++count;
		}
		sum += count;
	}
	printf("\n生成YUV序列，共%d帧。\n", sum);

	//释放空间
	free(rgbBuf);
	free(yBuf);
	free(uBuf);
	free(vBuf);
	fclose(bmpFile);
	fclose(yuvFile);
	return 0;
}

4.2 bmp2rgb.cpp

#include<stdlib.h>
#include<string.h>
#include"bmp2yuv.h"
int bmp2rgb(unsigned char* bbuf, int w, int h, unsigned char* rgbbuf)
{
	unsigned char* rgb = ((unsigned char*)malloc(w * h));

	for (int i = 0; i < h; i++)
	{
		for (int j = 0; j < w; j++)
		{
			rgb[i * w + j] = bbuf[(h - i - 1) * w + j];  //倒序读取
		}
	}
	memcpy(rgbbuf, rgb, w * h);//把倒叙后的缓冲区数据复制给输出缓冲区
	free(rgb);
	return 0;
}

4.3 rgb2yuv.cpp

#include<stdlib.h>
#include"bmp2yuv.h"

static float RGBYUV02990[256], RGBYUV05870[256], RGBYUV01140[256];
static float RGBYUV01684[256], RGBYUV03316[256];
static float RGBYUV04187[256], RGBYUV00813[256];

//查找表
void InitLookupTable()
{
	for (int i = 0; i < 256; i++)
	{
		RGBYUV02990[i] = (float)0.2990 * i;
		RGBYUV05870[i] = (float)0.5870 * i;
		RGBYUV01140[i] = (float)0.1140 * i;
		RGBYUV01684[i] = (float)0.1684 * i;
		RGBYUV03316[i] = (float)0.3316 * i;
		RGBYUV04187[i] = (float)0.4187 * i;
		RGBYUV00813[i] = (float)0.0813 * i;
	}
}

int rgb2yuv(unsigned long w, unsigned long h, unsigned char* rgbdata, unsigned char* y, unsigned char* u, unsigned char* v)
{
	InitLookupTable();//初始化查找表

	//定义缓冲区，4：2：0格式uv要下采样
	unsigned char* utemp = NULL;
	unsigned char* vtemp = NULL;
	utemp = (unsigned char*)malloc(w * h);
	vtemp = (unsigned char*)malloc(w * h);
	unsigned long i, r, g, b, nSize;

	//rgb转yuv
	for (i = 0, nSize = 0; nSize < w * h * 3; nSize += 3)
	{
		b = rgbdata[nSize];
		g = rgbdata[nSize + 1];
		r = rgbdata[nSize + 2];
		y[i] = (unsigned char)(RGBYUV02990[r] + RGBYUV05870[g] + RGBYUV01140[b]);
		utemp[i] = (unsigned char)(-RGBYUV01684[r] - RGBYUV03316[g] + b / 2 + 128);
		vtemp[i] = (unsigned char)(r / 2 - RGBYUV04187[g] - RGBYUV00813[b] + 128);
		i++;
	}

	// 4：4：4到4：2：0格式，u、v均为y的1/4,需进行下采样
	int k = 0;
	for (i = 0; i < h; i += 2)
	{
		for (unsigned long j = 0; j < w; j += 2)
		{
			u[k] = (utemp[i * w + j] + utemp[(i + 1) * w + j] + utemp[i * w + j + 1] + utemp[(i + 1) * w + j + 1]) / 4;
			v[k] = (vtemp[i * w + j] + vtemp[(i + 1) * w + j] + vtemp[i * w + j + 1] + vtemp[(i + 1) * w + j + 1]) / 4;
			k++;
		}
	}

	//限电平处理
	for (i = 0; i < w * h; i++)
	{
		if (y[i] < 16)
			y[i] = 16;
		if (y[i] > 235)
			y[i] = 235;
	}

	for (i = 0; i < w * h / 4; i++)
	{
		if (u[i] < 16)
			u[i] = 16;
		if (u[i] > 240)
			u[i] = 240;
		if (v[i] < 16)
			v[i] = 16;
		if (v[i] > 240)
			v[i] = 240;
	}
	if (utemp)
		free(utemp);
	if (vtemp)
		free(vtemp);
	return 0;
}

4.4 bmp2yuv.h

#pragma once
 
#include<windows.h>

int bmp2rgb(unsigned char* bbuf, int w, int h, unsigned char* rgbbuf);
void InitLookupTable();
int rgb2yuv(unsigned long w, unsigned long h, unsigned char* rgbdata, unsigned char* y, unsigned char* u, unsigned char* v);

实验结果：

输入参数 bmp文件帧数 yuv文件，生成yuv序列共250帧