基于颜色识别的火焰检测算法的opencv实现

#include "cv.h" #include "highgui.h" #include <stdio.h> #include <string.h> #include <math.h> int main(int argc,char** argv) { //变量定义 IplImage *image=0,*img=0, *yuv=0; int i,j; int height,width,step,channels; uchar* data; //CvMat* pFrMat=NULL; CvScalar scalar1; int VECT=0;//阈值参数设为35 char* filename=argc==2?argv[1]:(char*)"fire2.jpg"; //读入图像 image=cvLoadImage(filename,-1); if(!image) printf("Could not load image file\n"); yuv=cvCreateImage(cvGetSize(image),8,3); img=cvCreateImage(cvGetSize(image),8,3); //创建窗口,显示源图像 cvNamedWindow("SourceImage",CV_WINDOW_AUTOSIZE); cvNamedWindow("ResultImage",CV_WINDOW_AUTOSIZE); cvShowImage("SourceImage",image); //转化为YCrCb空间,获取图像信息 cvCvtColor(image,yuv,CV_BGR2YCrCb); height = yuv->height; width = yuv->width; step = yuv->widthStep; channels = yuv->nChannels; data = (uchar *)yuv->imageData; printf("Processing a %dx%d image with %d channels\n",height,width,channels); /************************测试得顺序为YCrCb********************************* /* IplImage *plane0=0,*plane1=0,*plane2=0; plane0=cvCreateImage(cvGetSize(image),8,1); plane1=cvCreateImage(cvGetSize(image),8,1); plane2=cvCreateImage(cvGetSize(image),8,1); cvCvtPixToPlane(yuv,plane0,plane1,plane2,0); cvNamedWindow("Image0",CV_WINDOW_AUTOSIZE); cvNamedWindow("Image1",CV_WINDOW_AUTOSIZE); cvNamedWindow("Image2",CV_WINDOW_AUTOSIZE); cvShowImage("Image0",plane0); cvShowImage("Image1",plane1); cvShowImage("Image2",plane2); cvWaitKey(); cvDestroyWindow("Image0"); cvDestroyWindow("Image1"); cvDestroyWindow("Image2"); */ //求YCrCb三量的均值 scalar1=cvAvg(yuv);//scalar1.val[0]是y通道均值,scalar1.val[1]是Cr通道均值 for(i=0;i<height;i++) for(j=0;j<width;j++) { if(data[i*step+j*channels+0]>data[i*step+j*channels+2]&&data[i*step+j*channels+0]>scalar1.val[0]&&\ data[i*step+j*channels+1]>data[i*step+j*channels+2]&&data[i*step+j*channels+1]>scalar1.val[1]&&\ data[i*step+j*channels+2]<scalar1.val[2]&&fabs(data[i*step+j*channels+1]-data[i*step+j*channels+2])>VECT) // if(data[0]>data[2] && data[0]>scalar1.val[0] && data[1]>data[2] && data[1]>scalar1.val[1] && data[2]<scalar1.val[2] && abs(data[1]-data[2])>VECT) { img->imageData[i*img->widthStep+j*img->nChannels+0]=255; img->imageData[i*img->widthStep+j*img->nChannels+1]=255; img->imageData[i*img->widthStep+j*img->nChannels+2]=255; } else { img->imageData[i*img->widthStep+j*img->nChannels+0]=0; img->imageData[i*img->widthStep+j*img->nChannels+1]=0; img->imageData[i*img->widthStep+j*img->nChannels+2]=0; } } cvShowImage("ResultImage",img); cvWaitKey(); cvDestroyWindow("SourceImage"); cvDestroyWindow("ResultImage"); }