YUV文件转RGB并保存为bmp格式

本文YUV文件以YUV420为例(YUV文件格式可以自行百度)。

方法一；显示在控件上没问题

#define unsigned char U8 bool Yv12ToBgb24(U8* pYUV, U8* pBGR24, int width, int height) { if (width < 1 || height < 1 || pYUV == NULL || pBGR24 == NULL) { return false; } const long len = width * height; U8* yData = pYUV; U8* vData = &yData[len]; U8* uData = &vData[len >> 2]; int bgr[3]; int yIdx, uIdx, vIdx, idx; for (int i = 0; i < height; i++) { for (int j = 0; j < width; j++) { yIdx = i * width + j; vIdx = (i / 2) * (width / 2) + (j / 2); uIdx = vIdx; bgr[0] = (int)(yData[yIdx] + 1.732446 * (uData[vIdx] - 128)); // b分量 bgr[1] = (int)(yData[yIdx] - 0.698001 * (uData[uIdx] - 128) - 0.703125 * (vData[vIdx] - 128)); // g分量 bgr[2] = (int)(yData[yIdx] + 1.370705 * (vData[uIdx] - 128)); // r分量 for (int k = 0; k < 3; k++) { idx = (i * width + j) * 3 + k; if (bgr[k] >= 0 && bgr[k] <= 255) { pBGR24[idx] = bgr[k]; } else { pBGR24[idx] = (bgr[k] < 0) ? 0 : 255; } } } } return true; }

方法二:保存在本地没问题

bool Yuv420ToRgb(U8* pYUV, U8* pRGB, int width, int height) { if (width < 1 || height < 1 || pYUV == NULL || pRGB == NULL) { return false; } //找到Y、U、V在内存中的首地址 U8* pY = pYUV; U8* pU = pYUV + height*width; U8* pV = pU + (height*width / 4); U8* pBGR = NULL; U8 R = 0; U8 G = 0; U8 B = 0; U8 Y = 0; U8 U = 0; U8 V = 0; double temp = 0; for (int i = 0; i < height; i++) { for (int j = 0; j < width; j++) { //找到相应的RGB首地址 pBGR = pRGB + i*width * 3 + j * 3; //取Y、U、V的数据值 Y = *(pY + i*width + j); U = *pU; V = *pV; //yuv转rgb公式 //yuv转rgb公式 temp = Y + ((1.773) * (U - 128)); B = temp < 0 ? 0 : (temp>255 ? 255 : (U8)temp); temp = (Y - (0.344) * (U - 128) - (0.714) * (V - 128)); G = temp < 0 ? 0 : (temp>255 ? 255 : (U8)temp); temp = (Y + (1.403)*(V - 128)); R = temp < 0 ? 0 : (temp>255 ? 255 : (U8)temp); //将转化后的rgb保存在rgb内存中，注意放入的顺序b是最低位 *pBGR = B; *(pBGR + 1) = G; *(pBGR + 2) = R; if (j % 2 != 0) { *pU++; *pV++; } } if (i % 2 == 0) { pU = pU - width / 2; pV = pV - width / 2; } } return true; }

保存bmp格式：

typedef struct MyBITMAPFILEHEADER_ST { unsigned int bfSize; unsigned short bfReserved1; unsigned short bfReserved2; unsigned int bfOffBits; }MyBITMAPFILEHEADER; typedef struct MyBITMAPINFOHEADER_ST { unsigned int biSize; int biWidth; int biHeight; unsigned short biPlanes; unsigned short biBitCount; unsigned int biCompression; unsigned int biSizeImage; int biXPelsPerMeter; int biYPelsPerMeter; unsigned int biClrUsed; unsigned int biClrImportant; }MyBITMAPINFOHEADER;

保存到本地

图片保存的时候，一定要注意bih.biHeight = -height; //图片如果颠倒，去掉负号

//保存到本地 void MySaveBmp(U8 *rgbbuf, int width, int height) { MyBITMAPFILEHEADER bfh; MyBITMAPINFOHEADER bih; unsigned short bfType = 0x4d42; bfh.bfReserved1 = 0; bfh.bfReserved2 = 0; bfh.bfSize = 2 + sizeof(MyBITMAPFILEHEADER) + sizeof(MyBITMAPINFOHEADER) + width*height * 3; bfh.bfOffBits = 0x36; bih.biSize = sizeof(MyBITMAPINFOHEADER); bih.biWidth = width; bih.biHeight = -height; //图片如果颠倒，去掉负号 bih.biPlanes = 1; bih.biBitCount = 24; bih.biCompression = 0; bih.biSizeImage = 0; bih.biXPelsPerMeter = 5000; bih.biYPelsPerMeter = 5000; bih.biClrUsed = 0; bih.biClrImportant = 0; FILE* file; fopen_s(&file, "test11.bmp", "wb"); if (!file) { //printf("Could not write file\n"); return; } fwrite(&bfType, sizeof(bfType), 1, file); fwrite(&bfh, sizeof(bfh), 1, file); fwrite(&bih, sizeof(bih), 1, file); fwrite(rgbbuf, width*height * 3, 1, file); fclose(file); }