C++实现YUV420转成BMP

已于 2024-01-19 22:43:27 修改
阅读量641 收藏 11
点赞数 8
分类专栏： 音视频 文章标签： c++ opencv 音视频
于 2024-01-19 21:54:31 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/sinat_38854292/article/details/135707506
版权

一、使用OpenCV库

在C++中，将YUV420数据转换为BMP图片需要一些图像处理的步骤。以下是一个简单的示例代码，使用了C++和OpenCV库来完成这个任务。在运行之前，请确保已经安装了OpenCV库：

#include <iostream>
#include <fstream>
#include <opencv2/opencv.hpp>

void YUV420toBGR(const unsigned char *yuv, int width, int height, cv::Mat &output) {
    cv::Mat yuvMat(height + height / 2, width, CV_8UC1, (void*)yuv);
    cv::cvtColor(yuvMat, output, cv::COLOR_YUV2BGR_I420);
}

int main() {
    // 读取YUV文件
    std::ifstream file("input.yuv", std::ios::binary | std::ios::ate);
    if (!file.is_open()) {
        std::cerr << "Error opening YUV file." << std::endl;
        return -1;
    }

    std::streamsize size = file.tellg();
    file.seekg(0, std::ios::beg);

    std::vector<char> buffer(size);
    if (!file.read(buffer.data(), size)) {
        std::cerr << "Error reading YUV file." << std::endl;
        return -1;
    }

    file.close();

    // 定义图像大小
    int width = 640;
    int height = 480;

    // 将YUV数据转换为BGR图像
    cv::Mat bgrImage;
    YUV420toBGR(reinterpret_cast<unsigned char*>(buffer.data()), width, height, bgrImage);

    // 保存BGR图像为BMP文件
    cv::imwrite("output.bmp", bgrImage);

    std::cout << "Conversion completed successfully." << std::endl;

    return 0;
}

在这个例子中，假设YUV数据的格式是I420。如果你的YUV数据格式不同，你可能需要调整cv::COLOR_YUV2BGR_I420这一行的参数。此外，请确保你的YUV文件的分辨率与代码中定义的分辨率一致。

二、不使用OpenCV库

如果不想使用OpenCV，可以直接处理YUV420数据并将其转换为BMP格式。以下是一个基本的示例，使用C++和标准库来完成这个任务：

#include <iostream>
#include <fstream>
#include <cstdint>
#include <vector>

// YUV to RGB conversion function
void YUV420toRGB(const unsigned char *yuv, int width, int height, std::vector<unsigned char> &rgb) {
    int size = width * height;
    int uvSize = size / 4;

    for (int i = 0; i < height; i++) {
        for (int j = 0; j < width; j++) {
            int y = yuv[i * width + j];
            int u = yuv[size + (i / 2) * (width / 2) + (j / 2)];
            int v = yuv[size + uvSize + (i / 2) * (width / 2) + (j / 2)];

            int c = y - 16;
            int d = u - 128;
            int e = v - 128;

            int r = (298 * c + 409 * e + 128) >> 8;
            int g = (298 * c - 100 * d - 208 * e + 128) >> 8;
            int b = (298 * c + 516 * d + 128) >> 8;

            r = (r < 0) ? 0 : ((r > 255) ? 255 : r);
            g = (g < 0) ? 0 : ((g > 255) ? 255 : g);
            b = (b < 0) ? 0 : ((b > 255) ? 255 : b);

            rgb.push_back(static_cast<unsigned char>(b));
            rgb.push_back(static_cast<unsigned char>(g));
            rgb.push_back(static_cast<unsigned char>(r));
        }
    }
}

// BMP file header structure
#pragma pack(push, 1)
struct BMPHeader {
    uint16_t signature;
    uint32_t fileSize;
    uint16_t reserved1;
    uint16_t reserved2;
    uint32_t dataOffset;
    uint32_t headerSize;
    int32_t width;
    int32_t height;
    uint16_t planes;
    uint16_t bitsPerPixel;
    uint32_t compression;
    uint32_t imageSize;
    int32_t xPixelsPerMeter;
    int32_t yPixelsPerMeter;
    uint32_t totalColors;
    uint32_t importantColors;
};
#pragma pack(pop)

int main() {
    // 读取YUV文件
    std::ifstream file("input.yuv", std::ios::binary | std::ios::ate);
    if (!file.is_open()) {
        std::cerr << "Error opening YUV file." << std::endl;
        return -1;
    }

    std::streamsize size = file.tellg();
    file.seekg(0, std::ios::beg);

    std::vector<char> buffer(size);
    if (!file.read(buffer.data(), size)) {
        std::cerr << "Error reading YUV file." << std::endl;
        return -1;
    }

    file.close();

    // 定义图像大小
    int width = 640;
    int height = 480;

    // 将YUV数据转换为RGB
    std::vector<unsigned char> rgbImage;
    YUV420toRGB(reinterpret_cast<unsigned char*>(buffer.data()), width, height, rgbImage);

    // 创建BMP文件头
    BMPHeader bmpHeader;
    bmpHeader.signature = 0x4D42; // BM
    bmpHeader.fileSize = sizeof(BMPHeader) + rgbImage.size();
    bmpHeader.reserved1 = 0;
    bmpHeader.reserved2 = 0;
    bmpHeader.dataOffset = sizeof(BMPHeader);
    bmpHeader.headerSize = 40;
    bmpHeader.width = width;
    bmpHeader.height = height;
    bmpHeader.planes = 1;
    bmpHeader.bitsPerPixel = 24;
    bmpHeader.compression = 0;
    bmpHeader.imageSize = 0;
    bmpHeader.xPixelsPerMeter = 0;
    bmpHeader.yPixelsPerMeter = 0;
    bmpHeader.totalColors = 0;
    bmpHeader.importantColors = 0;

    // 写入BMP文件
    std::ofstream bmpFile("output.bmp", std::ios::binary);
    bmpFile.write(reinterpret_cast<char*>(&bmpHeader), sizeof(BMPHeader));
    bmpFile.write(reinterpret_cast<char*>(rgbImage.data()), rgbImage.size());
    bmpFile.close();

    std::cout << "Conversion completed successfully." << std::endl;

    return 0;
}

如果转换出来的图片颜色不正确，可能是因为 YUV420 到 RGB 的转换公式不准确或者颜色通道的顺序不正确。修改代码如下：

void YUV420toRGB(const unsigned char* yuv, int width, int height, std::vector<unsigned char>& rgb) {
    int size = width * height;
    int uvSize = size / 4;

    for (int i = 0; i < height; i++) {
        for (int j = 0; j < width; j++) {
            int y = yuv[i * width + j];
            int u = yuv[size + (i / 2) * (width / 2) + (j / 2)];
            int v = yuv[size + uvSize + (i / 2) * (width / 2) + (j / 2)];

            // YUV to RGB conversion
            int c = y - 16;
            int d = u - 128;
            int e = v - 128;

            int r = (298 * c + 409 * e + 128) >> 8;
            int g = (298 * c - 100 * d - 208 * e + 128) >> 8;
            int b = (298 * c + 516 * d + 128) >> 8;

            r = (r < 0) ? 0 : ((r > 255) ? 255 : r);
            g = (g < 0) ? 0 : ((g > 255) ? 255 : g);
            b = (b < 0) ? 0 : ((b > 255) ? 255 : b);

            rgb.push_back(static_cast<unsigned char>(r));
            rgb.push_back(static_cast<unsigned char>(g));
            rgb.push_back(static_cast<unsigned char>(b));
        }
    }
}

如果图像是颠倒的，可能是由于 BMP 文件的行顺序与 YUV 数据的行顺序不匹配。在 BMP 文件中，图像的行通常是从底部到顶部排列的，而 YUV 数据的行可能是从顶部到底部排列的。因此，在写入 BMP 文件时，你需要按照适当的顺序写入行。修改代码如下：

// 写入BMP文件
std::ofstream bmpFile("output.bmp", std::ios::binary);
bmpFile.write(reinterpret_cast<char*>(&bmpHeader), sizeof(BMPHeader));

// 写入图像数据（按适当的行顺序）
for (int i = height - 1; i >= 0; i--) {
    bmpFile.write(reinterpret_cast<char*>(&rgbImage[i * width * 3]), width * 3);
}

bmpFile.close();

三、非YUV420

上述的YUV到BMP的转换代码是为YUV420格式设计的。在YUV420格式中，图像的亮度（Y分量）是全分辨率的，而色度（U和V分量）是以1/2的水平和垂直分辨率存储的。如果YUV数据格式不是YUV420，而是其他的YUV格式（比如YUV422、YUV444等），需要相应地修改颜色分量的提取和转换公式。以下是对于YUV422格式的简单修改，假设颜色分量的存储顺序为YUYV：

void YUV422toRGB(const unsigned char* yuv, int width, int height, std::vector<unsigned char>& rgb) {
    int size = width * height * 2;

    for (int i = 0; i < height; i++) {
        for (int j = 0; j < width; j += 2) {
            int y1 = yuv[i * width * 2 + j * 2];
            int u = yuv[i * width * 2 + j * 2 + 1];
            int y2 = yuv[i * width * 2 + j * 2 + 2];
            int v = yuv[i * width * 2 + j * 2 + 3];

            // YUV to RGB conversion for the first pixel
            // (similarly, you can convert the second pixel)
            int c1 = y1 - 16;
            int d1 = u - 128;
            int e1 = v - 128;

            int r1 = (298 * c1 + 409 * e1 + 128) >> 8;
            int g1 = (298 * c1 - 100 * d1 - 208 * e1 + 128) >> 8;
            int b1 = (298 * c1 + 516 * d1 + 128) >> 8;

            r1 = std::min(255, std::max(0, r1));
            g1 = std::min(255, std::max(0, g1));
            b1 = std::min(255, std::max(0, b1));

            // YUV to RGB conversion for the second pixel
            // (similarly, you can convert the first pixel)
            int c2 = y2 - 16;
            int d2 = u - 128;
            int e2 = v - 128;

            int r2 = (298 * c2 + 409 * e2 + 128) >> 8;
            int g2 = (298 * c2 - 100 * d2 - 208 * e2 + 128) >> 8;
            int b2 = (298 * c2 + 516 * d2 + 128) >> 8;

            r2 = std::min(255, std::max(0, r2));
            g2 = std::min(255, std::max(0, g2));
            b2 = std::min(255, std::max(0, b2));

            // Append the RGB values to the vector
            rgb.push_back(static_cast<unsigned char>(b1));
            rgb.push_back(static_cast<unsigned char>(g1));
            rgb.push_back(static_cast<unsigned char>(r1));

            rgb.push_back(static_cast<unsigned char>(b2));
            rgb.push_back(static_cast<unsigned char>(g2));
            rgb.push_back(static_cast<unsigned char>(r2));
        }
    }
}
心愿许得无限大
关注
专栏目录
Yuv转Bmp bmp转yuv420 支持各种图片格式转yuv
04-22
Yuv转Bmp bmp转yuv420 支持各种图片格式转yuv bmp转yuv420 y分量无像素点差异 色度分量有0-1个像素点差异
yuv2bmp(yuv转bmp)
04-14
简单使用的小工具，mfc实现yuv视频序列转BMP
YUV420存储为BMP和JPG图片
m0_60259116的博客
09-25 512
YUV420存储为BMP和JPG图片
yuv转bmp
aig8440的博客
04-15 106
#ifdef _INTERFACE_H #error _INTERFACE_H has be exsisted #else #define _INTERFACE_H #include "stdafx.h"//类型定义在这个文件中 #include "stdlib.h" #include "windows.h" void SaveYUV(AVFrame...
YUV420P转BMP指南
weixin_44948689的博客
04-19 1509
YUV420P转BMP指南 一、YUV像素点转换为RGB公式 R = Y + 1.403 * (U - 128) G = Y - 0.343 * (V - 128) - 0.714 * (U - 128) B = Y + 1.770 * (V - 128) 同样值得注意的是：网上的公式依然是反过来的（网上对应V的地方是U，U的地方是V） 二、从YUV420P格式到BMP转换算法 ​ 由上图可得，yuv420格式是每四个Y公用一对UV，所以U和V的数量都为Y的四分之一 ​ 而：下图则是YUV420在图像文件
c++实现NV21（YUV420SP）转BMP
qq_42955378的博客
06-02 890
YUV420有很多种格式，此脚本是将 YUV420SP （也可以称为 NV21）格式转换成BMP格式并储存。 最好将脚本与文件路径放在一个盘下，比如我这都在D盘，当文件在C盘，脚本在D盘运行时，会出现无法获取文件名的情况 已经在windows编译通过 #include <stdio.h> #include <stdlib.h> #include <vector> #include <iostream> #include <fstream>
bmp文件转yuv420
10-18
在计算机图形学和视频编码领域，BMP（Bitmap）是一种常见的位图文件格式，而YUV420则是一种用于视频编码的色彩空间。本文将详细介绍如何使用CUDA和DirectX编程技术将BMP图片转换为YUV420格式。 首先，让我们了解BMP...
yuv420数据保存为bmp图片
03-18
将预览的到的每一帧yuv420数据通过转码(yuv420->rgb565->bmp)，保存
YUV420转BMP图片 VC
04-28
这篇文档将详细解释如何使用C++语言将`YUV420`格式的数据转换成`BMP`图片，主要基于提供的`VC`（Visual C++）项目代码。 首先，理解`YUV420`格式至关重要。`YUV420`是一种色彩空间，将图像分为亮度（Y）和两个色度...
学习用C语言转换图片格式将bmp转换成yuv格式
05-24
YUV文件通常有不同格式，如YUV420p，其中Y分量占据全分辨率，而U和V分量是1/4分辨率。因此，写入文件时需要考虑这种布局。 5. **优化与效率**：为了提高效率，可以考虑使用内存映射（memory mapping）技术，将BMP...
YUV BMP两种格式相互转换
06-01
源码，可以将YUV转换成BMP，也可以将BMP转换为YUV，c#工程，但转换部分为纯c/c++实现，可以直接移植出。
yuv2bmp:将 YUV 数据转换为 BMP 格式的数据
06-20
yuv2bmp 将 YUV 数据转换为 BMP 格式的数据。
利用ffmpeg将yuv转为bmp格式
01-09
先利用ffmpeg对接收到的yuv视频流解码（不含接收和解码过程），然后转为bmp格式。
BMP YUV格式转化工具
06-14
bmp、yuv格式图像及 bmp转YUV工具，方便使用，支持bmp24位
YUV文件读取.rar_C++读取YUV_YUV_yuv；bmp_图片YUV_读取yuv
09-23
这个"YUV文件读取.rar"压缩包提供了一个C++实现，用于读取、处理和显示YUV格式的文件。以下是关于YUV格式、C++读取YUV以及与BMP图片转换的相关知识点： 1. **YUV颜色空间**： - YUV是一种分离亮度（Y）和色度（U、...
【数字图像处理】YUV420转RGB并BMP存储<纯C++实现>
zhengtu009的专栏
12-09 4804
1、读取akiyo_qcif.yuv YUV420文件，按帧读取，转RGB，并存储到BMP文件。2、暂时实现读取一帧并存储一张BMP图片。 若要读取YUV序列，写成循环方式即可。3、生成的BMP文件还有一点小问题：图像的倒立的，而且红色偏多。 YUV是qcif分辨率。 估计是YUV到RGB的计算公式有点问题。解决中。。。。 图像倒立问题已经解决。红色不知道是不是公式变换和RGB数值区间限制的
ffmpeg工具 - YUV422转BMP
最新发布
sz66cm 学习随笔
01-05 608
需要将YUV422的数据转BMP图片，可能是多帧合在一起a.yuv需要转成p0.bmp p1.bmp … pn.bmp。
yuv420 转换成 bmp
aiq8620的博客
07-02 442
源码如下： // ConsoleApplication1.cpp : Defines the entry point for the console application. #include "stdafx.h" #include <stdio.h> #include <stdlib.h> #define XSIZE 1920 #define YSI...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值