将yuv图像转为.jpg文件

使用c++ +opencv实现,将yuv序列转为.jpg文件保存。

# define _CRT_SECURE_NO_WARNINGS
#include <string>
#include <iostream>
#include <fstream>

#include <cv.h>  
#include <highgui.h> 

#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>

using namespace cv;
using namespace std;


int main()
{
    int iWidth;
    int iHeight;
    int iFrameNum;
    int iImageSize;

    iWidth = 640;
    iHeight = 480;
    char *inputFileName = "640x480_YUV400.yuv";

    FILE *fpIn;
    if (fopen_s(&fpIn, inputFileName, "rb"))
    {
        cout << "File Open Failed!\n";
        system("pause");
        exit(1);
    }

    iImageSize = iWidth * iHeight;

    unsigned char *InData = (unsigned char*)malloc(iImageSize * sizeof(unsigned char));
    unsigned char *uvData = (unsigned char*)malloc(iImageSize / 2 * sizeof(unsigned char));//uv
    memset(uvData, 128, iImageSize / 2);

    Mat  frameYUV(iHeight * 3 / 2, iWidth, CV_8UC1);
    Mat  frameBGR;
    Mat  frameRGB;
    Mat  frameYUV420;

    char outName[128];
    iFrameNum = 0;
    while (1)
    {
        size_t size = fread(InData, sizeof(unsigned char), iImageSize, fpIn);
        if (size == 0)
        {
            cout << "Read Frame Fail!\n";
            system("pause");
            break;
        }
        memcpy(frameYUV.data, InData, iImageSize);
        memcpy(frameYUV.data + iImageSize, uvData, iImageSize / 2);

        cvtColor(frameYUV, frameBGR, CV_YUV2BGR_I420);
        cvtColor(frameBGR, frameRGB, CV_BGR2RGB);

        imshow("video", frameRGB);
        waitKey(1);

        cout << iFrameNum++ << " Frame Processed\n";

        sprintf(outName, "outFile/%d.jpg", iFrameNum);
        imwrite(outName, frameRGB);

    }

    free(InData);
    free(uvData);
    fclose(fpIn);

    return 0;
}

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值