图像处理-RGB24转YUV420遇到的坑以及执行效率对比

最新推荐文章于 2024-08-11 08:45:06 发布
最新推荐文章于 2024-08-11 08:45:06 发布
阅读量6.1k 收藏 26
点赞数 5
分类专栏： 图像处理 文章标签： RGB转YUV420
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/armwind/article/details/99943987
版权

感兴趣可以加QQ群85486140，大家一起交流相互学习下！

文章目录

一、色彩空间简介及一些参考文档

不同的色彩空间，颜色的表现形式不同(见http://colorizer.org/　这个网站中简单介绍了各个色彩空间的模型。)。色彩空间中的颜色是可以相互转化的。我们常用的颜色空间有YUV和RGB色彩空间。

二、YUV和RGB转换公式

如果是RGB24的话，则它们和YUV的对应公式如下所示：

  • 1.RGB转YUV

Y = 0.299R + 0.587G + 0.114B
U = -0.147R - 0.289G + 0.436B
V = 0.615R - 0.515G - 0.100B

  • 2.YUV转RGB

R = Y + 1.14V
G = Y - 0.39U - 0.58V
B = Y + 2.03U

上面可以可以看到有浮点运算，这会导致耗时很长。下面的优化也是基于浮点运算，来的。

三、RGB和YUV420转换提前需要了解的知识

1.YUV420内存布局

                   W
         +--------------------+
         |Y0Y1Y2Y3...         |
         |...                 |
         |                    |   H
         |                    |
         |                    |
         |                    |
         +--------------------+
         |U0U1...   |
         |...       |   H/2
         |          |
         +----------+
         |V0V1...   |
         |...       |  H/2
         |          |
         +----------+
             w/2

2.RGB内存布局

虽然名字上叫RGB,但是排列的时候是按着BGR的顺序排列的，如下所示。

                        B                               G                               R
        +-----------------------------------------------------------------------------------------------+
  高字节 | B | B | B | B | B | B | B | B | G | G | G | G | G | G | G | G | R | R | R | R | R | R | R | R |   低字节
        +-----------------------------------------------------------------------------------------------+
          0   1   2   3   4   5   6   7   8   9  10  11  12  13  14  15  16  17  18  19  20  21  22  23

3.转换方式

转换的时候由于不是RGB24转YUV444(一对一)，而是转成YUV420.所以必然有些数据时要舍弃的。下面即将处理的图片分辨率是320 x 240的大小，所以RGB24的格式就会产生320 * 240 *3 byte的数据 即230400，下图也能看到后缀为rgb图片的大小。对应YUV420图片大小则为320 * 240 * 1.5=115200 Byte的大小。同样下面实际生成的图片也是这么大。
在这里插入图片描述
RGB的存放方式如下所示，例如8*6像素的RGB图像，其数据分布式这样存放的。

           pixe0       pixe1      pixe2       pixe3       pixe4      pixe5        pixe6       pixe7
       +-----------|-----------|-----------|-----------|-----------|-----------|-----------|-----------+
Line0  | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | 
       +-----------|-----------|-----------|-----------|-----------|-----------|-----------|-----------+
Line1  | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | 
       +-----------|-----------|-----------|-----------|-----------|-----------|-----------|-----------+
Line2  | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | 
       +-----------|-----------|-----------|-----------|-----------|-----------|-----------|-----------+
Line3  | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | 
       +-----------|-----------|-----------|-----------|-----------|-----------|-----------|-----------+
Line4  | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | 
       +-----------|-----------|-----------|-----------|-----------|-----------|-----------|-----------+
Line5  | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | B | G | R | 
       +-----------|-----------|-----------|-----------|-----------|-----------|-----------|-----------+

那么由于UV是每4个点对应一组UV,则选择哪个点转换为U，那个点转换为V。针对这样的block选择每个block的左上角(pix0)像素转换成U，左下角(pix2)的转换成V，其实这4个点的像素都可以用来转换UV，只是大家都习惯这样来转。

|pix0|pix1|
|pix2|pix3|

其它网友YUV420和YUV444相互转换的的好博客：https://blog.csdn.net/guyuealian/article/details/82454945

四、源代码执行效率对比

VS2015源码包可以在这里下载：https://download.csdn.net/download/armwind/11589508
由于原始的转换过程中包含了浮点和乘法运算，所以需要优化算法。下面例程中也包含了优化前后的对比。

  • 1.源码
#include "stdafx.h"
#include "iostream"
#include<Windows.h>
#include <stdio.h>

using namespace::std;

static void RGBToYUV_v1(int R, int G, int B, int* Y, int* U, int* V) {
	*Y = 0.299*R + 0.587*G + 0.114*B;
	*U = -0.147*R - 0.289*G + 0.436*B;
	*V = 0.615*R - 0.515*G - 0.100*B;
}

static void YUVToRGB_v1(int Y, int U, int V, int* R, int* G, int* B) {
	*R = Y + 1.14*V;
	*G = Y - 0.39*U - 0.58*V;
	*B = Y + 2.03*U;
}

static void RGBToYUV(int Red, int Green, int Blue, int* Y, int* U, int* V)

{
	*Y = ((Red << 6) + (Red << 3) + (Red << 2) + Red + (Green << 7) + (Green << 4) + (Green << 2) + (Green << 1) + (Blue << 4) + (Blue << 3) + (Blue << 2) + Blue) >> 8;
	*U = (-((Red << 5) + (Red << 2) + (Red << 1)) - ((Green << 6) + (Green << 3) + (Green << 1)) + ((Blue << 6) + (Blue << 5) + (Blue << 4))) >> 8;
	*V = ((Red << 7) + (Red << 4) + (Red << 3) + (Red << 2) + (Red << 1) - ((Green << 7) + (Green << 2)) - ((Blue << 4) + (Blue << 3) + (Blue << 1))) >> 8;
}

static void YUVToRGB(int Y, int U, int V, int* Red, int* Green, int* Blue)
{
	*Red = ((Y << 8) + ((V << 8) + (V << 5) + (V << 2))) >> 8;
	*Green = ((Y << 8) - ((U << 6) + (U << 5) + (U << 2)) - ((V << 7) + (V << 4) + (V << 2) + V)) >> 8;
	*Blue = ((Y << 8) + (U << 9) + (U << 3)) >> 8;
}
/*********************只不包含浮点运算********************/
//如果使用原始的转换出来的话，会出现偏色问题，下面在原始的基础上又做了微调，可以正常显示。
#define RGB2Y(r,g,b) \
	((unsigned char)((66 * r + 129 * g + 25 * b + 128) >> 8) + 16)
//	(((r << 6) + (r << 3) + (r << 2) + r + (g << 7) + (g << 4) + (g << 2) + (g << 1) + (b << 4) + (b << 3) + (b << 2) + b) >> 8)

#define RGB2U(r,g,b) \
	((unsigned char)((-38 * r - 74 * g + 112 * b + 128) >> 8) + 128)
//	((-((r << 5) + (r << 2) + (r << 1)) - ((g << 6) + (g << 3) + (g << 1)) + ((b << 6) + (b << 5) + (b << 4))) >> 8)

#define RGB2V(r,g,b) \
	((unsigned char)((112 * r - 94 * g - 18 * b + 128) >> 8) + 128)
//	(((r << 7) + (r << 4) + (r << 3) + (r << 2) + (r << 1) - ((g << 7) + (g << 2)) - ((b << 4) + (b << 3) + (b << 1))) >> 8)

/**********************不包含浮点和乘法运算*****************/
#define RGB2Y_SHIFT(r,g,b) \
	((unsigned char)((((r << 6) + (r << 3) + (r << 2) + r + (g << 7) + (g << 4) + (g << 2) + (g << 1) + (b << 4) + (b << 3) + (b << 2) + b) + 128) >> 8) + 16)

#define RGB2U_SHIFT(r,g,b) \
	((unsigned char)(((-((r << 5) + (r << 2) + (r << 1)) - ((g << 6) + (g << 3) + (g << 1)) + ((b << 6) + (b << 5) + (b << 4))) + 128) >> 8) + 128)

#define RGB2V_SHIFT(r,g,b) \
	((unsigned char)(((r << 7) + (r << 4) + (r << 3) + (r << 2) + (r << 1) - ((g << 7) + (g << 2)) - ((b << 4) + (b << 3) + (b << 1)) + 128) >> 8) + 128)

/*************************包含浮点和乘法运算，没有优化**************/

#define RGB2Y_FLOAT(r,g,b) \
	((unsigned char)(0.299*r + 0.587*g + 0.114*b))

#define RGB2U_FLOAT(r,g,b) \
	((unsigned char)(-0.147*r - 0.289*g + 0.436*b))

#define RGB2V_FLOAT(r,g,b) \
	((unsigned char)(0.615*r - 0.515*g - 0.100*b))

void rgb2yuv420_multip(unsigned char *rgb_buf, unsigned char *yuv_buf, int width, int heigh) {

	if (!rgb_buf || !yuv_buf || !width || !heigh) {
		printf("invalid param\n");
		return;
	}
	unsigned char *y = yuv_buf;

	unsigned char *u = yuv_buf + width*heigh;
	unsigned char *v = u + width*heigh/4;

	for (int r = 0; r < heigh; r++ ) {
		for (int c = 0; c < width; c++) {
			int index = r*width + c;
			*y++ = RGB2Y(rgb_buf[index * 3 + 2], rgb_buf[index * 3 + 1], rgb_buf[index * 3]);
			if((r%2==0)&&(c%2 ==0)) //偶数行，偶数列获取U
				*u++ = RGB2U(rgb_buf[index * 3 + 2], rgb_buf[index * 3 + 1], rgb_buf[index * 3]);
			else if(c%2 == 0) //奇数行，偶数列获取V
				*v++ = RGB2V(rgb_buf[index * 3 + 2], rgb_buf[index * 3 + 1], rgb_buf[index * 3]);
		}
	}

}

void rgb2yuv420_shift(unsigned char *rgb_buf, unsigned char *yuv_buf, int width, int heigh) {

	if (!rgb_buf || !yuv_buf || !width || !heigh) {
		printf("invalid param\n");
		return;
	}
	unsigned char *y = yuv_buf;

	unsigned char *u = yuv_buf + width*heigh;
	unsigned char *v = u + width*heigh/4;

	for (int r = 0; r < heigh; r++ ) {
		for (int c = 0; c < width; c++) {
			int index = r*width + c;
			*y++ = RGB2Y(rgb_buf[index * 3 + 2], rgb_buf[index * 3 + 1], rgb_buf[index * 3]);
			if((r%2==0)&&(c%2 ==0)) //偶数行，偶数列获取U
				*u++ = RGB2U(rgb_buf[index * 3 + 2], rgb_buf[index * 3 + 1], rgb_buf[index * 3]);
			else if(c%2 == 0) //奇数行，偶数列获取V
				*v++ = RGB2V(rgb_buf[index * 3 + 2], rgb_buf[index * 3 + 1], rgb_buf[index * 3]);
		}
	}

}


void rgb2yuv420_float(unsigned char *rgb_buf, unsigned char *yuv_buf, int width, int heigh) {

	if (!rgb_buf || !yuv_buf || !width || !heigh) {
		printf("invalid param\n");
		return;
	}
	unsigned char *y = yuv_buf;

	unsigned char *u = yuv_buf + width*heigh;
	unsigned char *v = u + width*heigh/4;

	for (int r = 0; r < heigh; r++ ) {
		for (int c = 0; c < width; c++) {
			int index = r*width + c;
			*y++ = RGB2Y_FLOAT(rgb_buf[index * 3 + 2], rgb_buf[index * 3 + 1], rgb_buf[index * 3]);
			if((r%2==0)&&(c%2 ==0)) //偶数行，偶数列获取U
				*u++ = RGB2U_FLOAT(rgb_buf[index * 3 + 2], rgb_buf[index * 3 + 1], rgb_buf[index * 3]);
			else if(c%2 == 0) //奇数行，偶数列获取V
				*v++ = RGB2V_FLOAT(rgb_buf[index * 3 + 2], rgb_buf[index * 3 + 1], rgb_buf[index * 3]);
		}
	}

}


int main(int arc, const char ** argv)
{
	int ret = 0;
	FILE *pFile = NULL,*pFile_out = NULL;
	int size = 0;
	int width = 0, heigh = 0;
	char path[] = "D:\\vs2015_project\\yuv22rgb\\girl320x240.yuv";

	unsigned char * rgb_buf = NULL;
	unsigned char *yuv_buf = nullptr;
	cout << "arc:" << arc << endl;
	if (arc == 1) {
		cout << "please input parameters!" << endl;
		return -1;
	}
	sscanf(argv[2],"%d", &width);
	sscanf(argv[3],"%d", &heigh);
	printf("file name:%s,width:%d,heigh:%d\n", argv[1], width, heigh);
	//cout << "file name:" << argv[1] << endl;

	pFile = fopen(argv[1], "rb");
	if (NULL == pFile) {
		cout << "open file failed!" << endl;
	}
	else {
		fseek(pFile, 0, SEEK_END);
		size = ftell(pFile);
		fseek(pFile, 0, SEEK_SET);
		cout << "file size is " << size << endl;
	}
	rgb_buf = (unsigned char *)malloc(size);
	if (NULL == rgb_buf) {
		cout << "malloc buffer failed, return" << endl;
		fclose(pFile);
		return -1;
	}
	int r_size = fread(rgb_buf, 1, size, pFile);
	if (r_size != size) {
		cout << "read error" << endl;
		ret = -1;
		goto release;
	}
	int yuv_size = width * heigh * 1.5;
	yuv_buf = (unsigned char*)malloc(yuv_size);
	if (NULL == yuv_buf)
		goto release;

	LARGE_INTEGER t1, t2,t3,t4;
	LARGE_INTEGER f;
	QueryPerformanceFrequency(&f); //得到CPU频率
	QueryPerformanceCounter(&t1); //得到开始时间

	rgb2yuv420_multip(rgb_buf, yuv_buf, width, heigh);
	QueryPerformanceCounter(&t2); //得到优化了浮点运算的计算时间
	double total1_time = (t2.QuadPart - t1.QuadPart) / (double)f.QuadPart; //单位秒

	rgb2yuv420_shift(rgb_buf, yuv_buf, width, heigh);
	QueryPerformanceCounter(&t3); //得到优化浮点和乘法运算之后结束时间
	double total2_time = (t3.QuadPart - t2.QuadPart) / (double)f.QuadPart; //单位秒

	rgb2yuv420_float(rgb_buf, yuv_buf, width, heigh);
	QueryPerformanceCounter(&t4); //原始没有优化的图像

	double total3_time_float = (t4.QuadPart - t3.QuadPart) / (double)f.QuadPart; //单位秒
	printf("shift_time:%f s,multip_time:%f s,time_float:%f s\n", total2_time,total1_time,total3_time_float);

	pFile_out = fopen(path, "wb+");
	if (pFile_out == nullptr) {
		printf("null fd\n");
		goto release;
	}
	printf("yuv_size:%d\n", yuv_size);
	int w_count = fwrite(yuv_buf, 1, yuv_size, pFile_out);
	printf("write_count:%d\n", w_count);
	fclose(pFile_out);
	
release:
	if (rgb_buf) {
		free(rgb_buf);
		rgb_buf = nullptr;
	}
	if (yuv_buf) {
		free(yuv_buf);
		yuv_buf = nullptr;
	}
	fclose(pFile);
    return 0;
}

运行结果如下所示：

D:\vs2015_project\yuv22rgb\yuv22rgb\Debug>yuv22rgb.exe …\girl320x240.rgb 320 240
arc:4
file name:…\girl320x240.rgb,width:320,heigh:240
file size is 230400
shift_time:0.000367 s,multip_time:0.000366 s,time_float:0.000593 s
yuv_size:115200
write_count:115200

从上面能够发现在同样的运行环境下，处理时间优化前后差异很大，其中优化移位操作，优化的时间不是很大，不知道跑在arm架构上时间会是怎么样的，后面会尝试一下。

优化项时间(单位S)
没有任何优化0.000593
优化浮点计算0.000367
优化浮点和乘法计算0.000366
  • 2.CPU硬件信息
    在这里插入图片描述
  • 3.图片对比,下图左边是原始的图片，右边是转换之后的YUV，可以发现转换之后的图片有点偏红，不过色调基本一致。
    在这里插入图片描述

五、编译过程中遇到的问题

  • 1.读取文件识别
    打开文件时应该加上“rb"模式
  • 2.写入文件有问题(写入文件的大小和预期不一致)
    写入文件时应该使用”wb+"
  • 3.图像偏色
    uv反色或者公式有问题
armwind
关注
专栏目录
YUV 和 RGB 颜色互转的小工具
11-21
YUV 和 RGB 颜色互转的小工具 自己写的，最近用到了YUV色彩，网上很多算法，可是都不甚完美，花了两天才弄明天，写个小工具，内涵源码，方便测试和使用。
RGB&YUV图像视频格式
IT菜鸟
10-07 1817
OpenCV中如果是单通道或者三通道一样的大的图像，直接使用copyTo()函数即可。对于yuv420的图像则不是适用。
色彩转换系列之RGB格式与YUV格式互转原理及实现
最新发布
生活需要深度
08-11 786
在现代彩色电视系统中，通常采用三管彩色摄像机或彩色CCD摄影机进行取像，然后把取得的彩色图像信号经分色、分别放大校正后得到RGB，再经过矩阵变换电路得到亮度信号Y和两个色差信号R-Y(即U)、B-Y(即V)，最后发送端将亮度和两个色差总共三个信号分别进行编码，用同一信道发送出去。采用YUV色彩空间的重要性是它的亮度信号Y和色度信号U、V是分离的。至此，RGB2YUV的转换公式就优化完毕了，这个优化，在移动端，速度会有很大的提高，至于一些测试数据，我就不列举了，只给个效果图吧，大家可以直接试一下就知道了。
[数据压缩]RGB文件转换YUV420文件
lii_by的博客
03-23 6668
RGB文件与YUV文件格式 RGB文件格式 RGB三个分量按照B、G、R的顺序储存（4:4:4） YUV文件格式 此次选用YUV420文件格式，每四个Y共用一组U、V分量（4:2:0） RGB到YUV的转换公式 Y = 0.299 R + 0.587 G + 0.114 B U = - 0.1687 R - 0.3313 G + 0.5 B + 128 V = 0.5 R - 0.4187 G - 0.0813 B + 128 c++代码实现 #include <.
rgb转yuv420源代码(C语言)
07-02
代码是用C语言写的，实现rgb24格式转yuv420格式，代码是项目源代码里一个单独的模块，本人一直在用，供大家参考。
算法优化:rgb向yuv的转化最优算法,快得让你吃惊(转)
weixin_33704234的博客
03-28 364
朋友曾经给我推荐了一个有关代码优化的pdf文档《让你的软件飞起来》，看完之后，感受颇深。为了推广其，同时也为了自己加深印象，故将其总结为word文档。下面就是其的详细内容总结，希望能于己于人都有所帮助。   速度取决于算法 同样的事情，方法不一样，效果也不一样。比如，汽车引擎，可以让你的速度超越马车，却无法超越音速；涡轮引擎，可以轻松超越音障，却无法飞出地球；如果有火箭发动机，就可以到达火星...
RGB24转YUV420P
lcalqf的专栏
04-05 7719
第一种方法，直接用算法转换，不用第三方库。注意BMP图像数据是倒着存储的，即从左下角开始。
rgb转yuv420，yuv444
01-18
用来python2的opencv接口，需要事先安装opencv for python2.7
RGB24与YUV420格式相互转换
07-14
具体实现了RGB24与YUV420格式相互转换，达到视频文件压缩目的。
rgb24转yuv420高效率 完整代码，C++实现
08-25
RGB24到YUV420的转换是图像处理领域中的常见操作，特别是在视频编码和解码中。这种转换涉及到将RGB色彩空间的数据转换为YUV色彩空间，以节省存储空间和提高编码效率。在C++中实现这个过程，我们可以从以下几个方面来...
YUV-RGB.rar_RGB YUV_RGB YUV_RGB-gt;YUV_yuv to rgb_yuvrgb转换dsp
09-24
综上所述，了解YUV到RGB的转换原理和实践是图像处理和数字视频工作中的基本技能。在实际应用中，这种转换对于确保图像在各种平台上的准确显示具有重要意义。通过掌握这些知识点，我们可以更好地理解和处理涉及不同...
rgb格式转换到yuv格式
10-30
bmp图片转换yuv图片，rgb格式转换到yuv420格式。matlab代码。
444_rgb_rgb888图像格式_rgb_RGB888对比YUV444_YCbCr444转RGB888_ycbcr转rgb8
10-02
这里我们主要探讨的是RGB888图像格式与YUV444和YCbCr444颜色空间之间的转换，以及如何将YCbCr444转换为RGB888。 RGB888是一种常见的颜色表示方式，它使用三个通道（红、绿、蓝）来表示每个像素的颜色，每个通道用8...
RGB转YUV：RGB转YUV-matlab开发
05-30
RGB到YUV颜色空间转换是图像处理中的常见操作，尤其在视频编码和显示系统中广泛使用。YUV颜色模型主要用于优化存储和传输效率，因为它更符合人类视觉系统的特性。MATLAB作为一种强大的数值计算和可视化环境，是进行...
rgb24转yuv420
08-25
RGB24到YUV420的转换是图像处理领域中的常见操作，特别是在视频编码和解码、图像显示以及跨平台多媒体应用中。RGB24是一种颜色空间，它使用3个8位字节来表示每个像素的红、绿、蓝分量，总共24位。而YUV420则是一种...
win32截屏并rgb24转yuv420
smwhotjay的专栏
09-23 417
void ScreenCap(void* buf, int w, int h) { HWND hDesk = GetDesktopWindow(); HDC hScreen = GetDC(hDesk); int width = w;//GetDeviceCaps(hScreen, HORZRES); int height = h;//GetDeviceC...
rgb24转yuv420P
急支糖浆的博客
02-14 1533
本文记录了将rgb24的像素数据转为yuv420P的像素数据的方法 1.直接使用公式，遍历rgb每个像素点并将rgb格式转换为yuv格式 static void rgb2yuv420(int w, int h, uint8_t *rgb, uint8_t *yuv) { int pixsize; int pixIndex; uint8_t *y,*u,*v; int...
python----RGB to YUV420
qq_43508270的博客
10-22 2670
python----RGB to YUV420
RGB位图转YUV420
wzj_whut的专栏
12-25 1740
本代码只是为了生成YUV数据用于测试. 以下代码有多处可以再优化 uint8_t* yuvArray(uint8_t* rgb, int w, int h) //+ (unsigned char*)yuvArray: (uint8_t*) rgb width:(int) w height:(int) h //这是objective-c的写法 { int pixelCount = w*h;...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值