python

超轻量级php框架startmvc

YUV转为jpg图像的实现

更新时间:2020-08-13 03:48:01 作者:startmvc
调用opencv库,将yuv图像转为jpg图像。代码如下:#define_CRT_SECURE_NO_WARNINGS#include<string>#incl

调用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图像的实现就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。

YUV jpg图像