1. 程式人生 > 程式設計 >YUV轉為jpg影象的實現

YUV轉為jpg影象的實現

呼叫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;
}

以上這篇YUV轉為jpg影象的實現就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。