avi视频格式转yuv格式与播放yuv视频

「爱情、让人受尽委屈。」 2022-08-02 14:42 321阅读 0赞

由于要用到yuv格式视频，而目前只有avi格式的视频，所以需要转换，并且opencv不支持yuv编码的视频播放，所以需要转换为rgb编码，而后播放。写了两个程序，以供参考：

1，avi格式视频转yuv格式视频

2，播放并保存yuv视频

#include <stdlib.h>
    #include <stdio.h>
    #include <math.h>
    #include <cv.h>
    #include <highgui.h>
    void avi2WriteYuv(const char *in,const char *out)
    {
    	cv::VideoCapture vc;
    	bool flag = vc.open(in);
    	if (!flag)
    	{
    		printf("avi file open error \n");
    		system("pause");
    		exit(-1);
    	}
    
    	int frmCount = vc.get(CV_CAP_PROP_FRAME_COUNT);
    	frmCount -= 5;
    	printf("frmCount: %d \n", frmCount);
    
    	int w = vc.get(CV_CAP_PROP_FRAME_WIDTH);
    	int h = vc.get(CV_CAP_PROP_FRAME_HEIGHT);
    	printf("wwwwww%d\n", w);
    	printf("hhhhhh%d\n", h);
    	int bufLen = w*h * 3 / 2;
    	unsigned char* pYuvBuf = new unsigned char[bufLen];
    	FILE* pFileOut = fopen(out, "w+");
    	if (!pFileOut)
    	{
    		printf("pFileOut open error \n");
    		system("pause");
    		exit(-1);
    	}
    	printf("pFileOut open ok \n");
    
    	for (int i = 0; i<frmCount; i++)
    	{
    		printf("%d/%d \n", i + 1, frmCount);
    
    		cv::Mat srcImg;
    		vc >> srcImg;
    
    		cv::imshow("img", srcImg);
    		cv::waitKey(1);
    
    		cv::Mat yuvImg;
    		cv::cvtColor(srcImg, yuvImg, CV_BGR2YUV_I420);
    		memcpy(pYuvBuf, yuvImg.data, bufLen*sizeof(unsigned char));
    
    		fwrite(pYuvBuf, bufLen*sizeof(unsigned char), 1, pFileOut);
    	}
    
    	fclose(pFileOut);
    	delete[] pYuvBuf;
    }
    void DisplayYUV2RGB(const char *dir,const char *in,int _w,int _h)
    {
    	int w = _w;
    	int h = _h;
    	printf("yuv file w: %d, h: %d \n", w, h);
    
    	FILE* pFileIn = fopen(in, "rb+");
    	int bufLen = w*h * 3 / 2;
    	unsigned char* pYuvBuf = new unsigned char[bufLen];
    	int iCount = 0;
    
    
    	for (int i = 0; i<95; i++)
    	{
    		fread(pYuvBuf, bufLen*sizeof(unsigned char), 1, pFileIn);
    
    		cv::Mat yuvImg;
    		yuvImg.create(h * 3 / 2, w, CV_8UC1);
    		memcpy(yuvImg.data, pYuvBuf, bufLen*sizeof(unsigned char));
    		cv::Mat rgbImg;
    		cv::cvtColor(yuvImg, rgbImg, CV_YUV2BGR_I420);
    
    		cv::imshow("img", rgbImg);
    		char s[100];
    		sprintf(s,"%spic%d%s",dir,i,".jpg");
    		cv::imwrite(s, rgbImg);
    		cv::waitKey(1);
    
    		printf("%d \n", iCount++);
    	}
    
    	delete[] pYuvBuf;
    
    
    	fclose(pFileIn);
    }
    
    int main(int argc, char *argv[])
    {
    	const char *in = "C:/Users/jiang/Desktop/output/outfile.avi";
    	const char *out = "C:/Users/jiang/Desktop/output/outfile.yuv";
    	const char *dir = "C:/Users/jiang/Desktop/output/tupian1/";
    	avi2WriteYuv(in,out);
    	DisplayYUVtoRGB(dir,out, 1024, 768);
    	return 0;
    }