海康網絡攝像機YV12轉換為BGR,由opencv Mat顯示 (轉)


我使用的是海康DS-2CD852MF-E, 200萬,網絡攝像機,已經比較老了,不過SDK在海康官網下載的,開發流程都差不多.

海康攝像機回調解碼后的視頻數據格式為YV12,順便說一下YV12的數據格式   YYYY  V U.   我這個是720P,即1280 * 720分辨率. 那么Y分量的數量為 1280 * 720 = 921600 字節, V = U 數量為Y的1/4, 即921600 / 4 =   230400字節,所以一幀720P的YV12數據量為1382400字節,與斷點調試查看nSize一致.

這篇博客實現方法均來自網上找的代碼,不是自己寫的.僅作為一個記錄.有一篇博客YUV與RGB格式轉換 里面就是YV12轉到BGR的實現,看着它的代碼改的.

1.用OpenCV  顏色空間轉換函數

Mat pImgYCrCb, pImg;
pImgYCrCb.create(720, 1280,CV_8UC3); pImg.create(720, 1280,CV_8UC3); yv12toYUV(pImgYCrCb.data, (uchar*)pBuf, pFrameInfo->nWidth,pFrameInfo->nHeight,pImgYCrCb.step); cvtColor(pImgYCrCb, pImg, CV_YCrCb2RGB);//轉化為RGB圖像,pImg是輸出的幀
void yv12toYUV(uchar *outYuv, uchar *inYv12, int width, int height,int widthStep)   {   int col,row;   unsigned int Y,U,V;   int tmp;   int idx;   for (row=0; row<height; row++)   {   idx=row * widthStep;   int rowptr=row*width;   for (col=0; col<width; col++)   {   tmp = (row/2)*(width/2)+(col/2);   Y=(unsigned int) inYv12[row*width+col];   U=(unsigned int) inYv12[width*height+width*height/4+tmp];   V=(unsigned int) inYv12[width*height+tmp];   outYuv[idx+col*3]   = Y;   outYuv[idx+col*3+1] = U;   outYuv[idx+col*3+2] = V;   }   }   }

2.另一種opencv cvtColor實現,較為簡單

Mat dst(pFrameInfo->nHeight,pFrameInfo->nWidth,CV_8UC3);//這里nHeight為720,nWidth為1280,8UC3表示8bit uchar 無符號類型,3通道值 Mat src(pFrameInfo->nHeight + pFrameInfo->nHeight/2,pFrameInfo->nWidth,CV_8UC1,(uchar*)pBuf); cvtColor(src,dst,CV_YUV2BGR_YV12); imshow("bgr",dst); waitKey(1);

3.只保留Y分量,觀看黑白視頻

uchar Y[921600];
memcpy(Y, pBuf, 921600); if (Videofile==NULL) { sprintf_s(filename,"is_Y.yuv",0); fopen_s(&Videofile,filename,"wb"); } fwrite(Y,921600,1,Videofile);

 

4.ffmpeg轉換

int sizeRGB = pFrameInfo->nWidth * pFrameInfo->nHeight * 3;   uchar* pBGR24 = new unsigned char[sizeRGB];//開辟存儲一副圖像大小的內存空間,因為這里是RGB圖像,所以需要 1280 * 720 * 3的大小 memset(pBGR24, 0, sizeRGB);  YV12ToBGR24_FFmpeg((uchar*)pBuf, pBGR24,1280, 720); Mat BGR24_IMG(pFrameInfo->nHeight,pFrameInfo->nWidth,CV_8UC3,pBGR24); imshow("bgr24", BGR24_IMG); waitKey(1); delete []pBGR24;

 

bool YV12ToBGR24_FFmpeg(unsigned char* pYUV,unsigned char* pBGR24,int width,int height) {     if (width < 1 || height < 1 || pYUV == NULL || pBGR24 == NULL)         return false;     AVPicture pFrameYUV,pFrameBGR;     avpicture_fill(&pFrameYUV,pYUV,AV_PIX_FMT_YUV420P,width,height);     //U,V互換     uint8_t * ptmp=pFrameYUV.data[1];     pFrameYUV.data[1]=pFrameYUV.data[2];     pFrameYUV.data [2]=ptmp;     avpicture_fill(&pFrameBGR,pBGR24,AV_PIX_FMT_BGR24,width,height);     struct SwsContext* imgCtx = NULL;     imgCtx = sws_getContext(width,height,AV_PIX_FMT_YUV420P,width,height,AV_PIX_FMT_BGR24,SWS_BILINEAR,0,0,0);     if (imgCtx != NULL){         sws_scale(imgCtx,pFrameYUV.data,pFrameYUV.linesize,0,height,pFrameBGR.data,pFrameBGR.linesize);         if(imgCtx){             sws_freeContext(imgCtx);             imgCtx = NULL;         }         return true;     }     else{         sws_freeContext(imgCtx);         imgCtx = NULL;         return false;     } }

注意這里ffmpeg會出現函數被否決,需要用新接口替換,也可以忽略這個警告,並且頭文件需要用extern "C"包含

extern "C" { #include <libavcodec\avcodec.h> #include <libswscale\swscale.h> #include <libavutil\pixfmt.h> #include <libavutil\imgutils.h> } #pragma   warning(disable: 4996)//忽略被否決的警告


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM