python 读取jpg文件是yuv_YUV转为jpg图像的实现

调用opencv库，将yuv图像转为jpg图像。

代码如下：

# define _CRT_SECURE_NO_WARNINGS

#include

#include

#include

#include

#include

#include

#include

#include

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