读取yuv并转化为RGB数据在Opencv中打开显示

来源：互联网发布：淘宝女士丝巾编辑：程序博客网时间：2024/05/01 05:21

/**

* Opencv中一般是读取 demo.avi 格式视频，下面程序是直接读取 demo.yuv 然后使用 Mat 保存

* 再使用 cv::cvtColor() 将每帧的 yuv 数据转化为 rgb 然后显示

#include <opencv/cv.h>#include <fstream>#include <iostream>#include <opencv2/core/core.hpp>#include <opencv2/highgui/highgui.hpp>#include <opencv2/imgproc/imgproc.hpp>using namespace std;using namespace cv;const int width = 1280;const int height = 720;const int framesize = width * height * 3 / 2;   //一副图所含的像素个数typedef struct planet{char name[framesize];double population;double g;} PLANET;int main(){/////////////////////////////////////////////////////////////// 计算视频的帧数，怎样替换成c语言形式的？PLANET pl;ifstream fin;fin.open("demo.yuv", ios_base::in|ios_base::binary);if(fin.fail()){cout << "the file is error" << endl;return -1;}fin.seekg(0, ios::end);   //设置文件指针到文件流的尾部streampos ps = fin.tellg();  //指出当前的文件指针unsigned long NumberPixe = ps;cout << "file size: " << ps << endl;  //输出指针的位置unsigned FrameCount = ps / framesize; //帧大小cout << "frameNuber: " << FrameCount; //输出帧数fin.close();/////////////////////////////////////////////////////////////////FILE* fileIn = fopen("demo.yuv", "rb+");unsigned char* pYuvBuf = new unsigned char[framesize]; //一帧数据大小//存储到图像namedWindow("yuv",  1);for(int i = 0; i < FrameCount; ++i){fread(pYuvBuf, framesize*sizeof(unsigned char), 1, fileIn);cv::Mat yuvImg;yuvImg.create(height*3/2, width, CV_8UC1);memcpy(yuvImg.data, pYuvBuf, framesize*sizeof(unsigned char));cv::Mat rgbImg; cv::cvtColor(yuvImg, rgbImg, CV_YUV2BGR_I420);cv::imshow("yuv", yuvImg); //只显示y分量cv::imshow("rgbImg", rgbImg);printf("第 %d 帧\n", i);int c = waitKey(30);if((char)c == 27){break;}}fclose(fileIn);cvDestroyWindow("yuv");cin.get();cin.get();return 0;}

0 0