海康網路攝像機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)//忽略被否決的警告