sz.zip_zip

#include "cv.h" #include "cxcore.h" #include "highgui.h" UCHAR getGray(IplImage *im,int x, int y){ return im->imageData[y*im->widthStep+x]; } void setGray(IplImage *im,int x,int y,UCHAR g){ im->imageData[y*im->widthStep+x]=g; } void FindSpot(IplImage* im,CvPoint p0,CvPoint &pbf,CvPoint &pja){ pbf=p0; pja=p0; CvPoint *oldlist=(CvPoint*)malloc(sizeof(CvPoint)); CvPoint *newlist; setGray(im,p0.x,p0.y,100); int oldcount=0,newcount=0; oldlist[oldcount++]=p0; do{ newlist=(CvPoint*)malloc(4*oldcount*sizeof(CvPoint)); newcount=0; for(int i=0;i<oldcount;++i){ CvPoint p=oldlist[i]; if(p.x>0){ if(getGray(im,p.x-1,p.y)==255) { newlist[newcount++]=cvPoint(p.x-1,p.y); setGray(im,p.x-1,p.y,100); if(pbf.x>p.x-1) pbf.x=p.x-1; } } if(p.x<im->width-1){ if(getGray(im,p.x+1,p.y)==255){ newlist[newcount++]=cvPoint(p.x+1,p.y); setGray(im,p.x+1,p.y,100); if(pja.x<p.x+1) pja.x=p.x+1; } } if(p.y>0){ if(getGray(im,p.x,p.y-1)==255){ newlist[newcount++]=cvPoint(p.x,p.y-1); setGray(im,p.x,p.y-1,100); if(pbf.y>p.y-1) pbf.y=p.y-1; } } if(p.y<im->height-1){ if(getGray(im,p.x,p.y+1)==255){ newlist[newcount++]=cvPoint(p.x,p.y+1); setGray(im,p.x,p.y+1,100); if(pja.y<p.y+1) pja.y=p.y+1; } } } free(oldlist); oldlist=newlist; oldcount=newcount; }while(newcount>0); free(newlist); } int main(){ const int n=10; CvCapture *src; IplImage *imH; src = cvCaptureFromAVI("aut1.avi"); IplImage *imV,*imX; for(int i=0;i<n;i++) { imV=cvQueryFrame(src); if(i==0){ imH=cvCloneImage(imV); } else{ cvAddWeighted(imH,(double)i/(i+1.0),imV,1.0/(i+1.0),0,imH); } } cvNamedWindow("Ablak",1); cvShowImage("Ablak",imH); /*IplConvKernel* bk = cvCreateStructuringElementEx(9,9,4,4,CV_SHAPE_RECT); IplConvKernel* bn = cvCreateStructuringElementEx(19,19,9,9,CV_SHAPE_RECT);*/ do { imV = cvQueryFrame(src); if(imV!=NULL) { /*cvNamedWindow("Film",1); cvShowImage("Film",imV); cvWaitKey(10);*/ imX=cvCloneImage(imV); IplImage *imA=cvCloneImage(imX); IplImage *imK=cvCloneImage(imX); cvSub(imX,imH,imA); cvSub(imH,imX,imK); cvAdd(imA,imK,imK); IplImage *imR=cvCreateImage(cvGetSize(imK),imK->depth,1); IplImage *imB=cvCloneImage(imR); IplImage *imG=cvCloneImage(imR); cvSplit(imK,imB,imG,imR,NULL); cvAdd(imB,imR,imR); cvAdd(imG,imR,imR); cvFlip(imR,imR); //cvSmooth(imR,imR,CV_BLUR,7,7); /*cvDilate(imR,imR,bk); cvErode(imR,imR,bk); cvErode(imR,imR,bn); cvDilate(imR,imR,bn); cvCmpS(imR,100,imR,CV_CMP_GE);*/ cvSmooth(imR,imR,CV_CMP_GE); cvCmpS(imR,100,imR,CV_CMP_GE); IplConvKernel *bk=cvCreateStructuringElementEx(9,9,4,4,CV_SHAPE_RECT); IplConvKernel *bn=cvCreateStructuringElementEx(19,19,9,9,CV_SHAPE_RECT); cvErode(imR,imR,bk,1); cvDilate(imR,imR,bk,1); cvDilate(imR,imR,bn,1); cvErode(imR,imR,bn,1); cvCopy(imV,imX); double maxval; do{ CvPoint pmax,pbf,pja; cvMinMaxLoc(imR,NULL,&maxval,NULL,&pmax); if(maxval>254.5){ FindSpot(imR,pmax,pbf,pja); if(pja.x+pbf.x+pja.y-pbf.y>50) cvRectangle(imX,pbf,pja,cvScalar(0,0,255,0),4); // piros teglalap rajzolasa a pont korul vonalvastagsag 4px } }while(maxval>254.5); cvNamedWindow("Film",1); cvShowImage("Film",imX); cvWaitKey(10); cvReleaseImage(&imR); cvReleaseImage(&imB); cvReleaseImage(&imG); cvReleaseImage(&imA); cvReleaseImage(&imK); cvReleaseImage(&imX); } } while(imV); }