Opencv抠图.doc

#include stdafx.h#include cv.h#include highgui.h#include cxcore.h#include #include using namespace std;#define pi 3.1415926#define h 20 /#define width 512#define height 480struct colornode int colorR; int colorG; int colorB;vector background;vector target;int flag;/1代表前景，2代表背景CvPoint prev_pt = -1,-1;IplImage *pImg,*newImg,*showImg;int color;IplImage* inpaint_mask = 0;IplImage* img0 = 0, *img = 0, *inpainted = 0;uchar *data,*newdata,*showdata;int channels, step,depth; double prbxheightwidth,prfxheightwidth;void on_mouse(int event,int x,int y,int flags,void * param) colornode a; if( event = CV_EVENT_LBUTTONUP | !(flags & CV_EVENT_FLAG_LBUTTON) ) prev_pt = cvPoint(-1,-1); else if( event = CV_EVENT_LBUTTONDOWN ) prev_pt = cvPoint(x,y); else if( event = CV_EVENT_MOUSEMOVE & (flags & CV_EVENT_FLAG_LBUTTON) CvPoint pt = cvPoint(x,y); if( prev_pt.x height; /width=pImg-width; newImg=cvCloneImage(pImg); showImg=cvCloneImage(pImg); cvNamedWindow(Interaction,-1); cvShowImage(Interaction,newImg); depth=pImg-depth; step=pImg-widthStep; channels=pImg-nChannels; data=(uchar *) pImg-imageData; newdata=(uchar *) newImg-imageData; showdata=(uchar *) showImg-imageData; cvSetMouseCallback( Interaction, on_mouse, 0); printf( Hot keys: n t ESC - quit the programn t 1 - target paintingn t 2 - background paintingn); for(;) char ch=cvWaitKey(); if (ch=3) break; if (ch=1) flag=1; if(ch= 2) flag=2; /cvShowImage( Interaction, newImg); /cvWaitKey(0); /继续处理 int sizeback,sizetarget; double distance; /data=(uchar *) pImg-imageData; for(i=0;iheight;i+) for(j=0;jwidth;j+) prbxij=0; prfxij=0;/target sizeback=background.size(); sizetarget=target.size(); printf(computing probability.n); for(i=0;iheight;i+) printf(i=%dn,i); for(j=0;jwidth;j+) for(k=0;ksizeback;k+) distance=(backgroundk.colorR-datai*step+j*channels)*(backgroundk.colorR-datai*step+j*channels)+(backgroundk.colorG-datai*step+j*channels+1)*(backgroundk.colorG-datai*step+j*channels+1)+(backgroundk.colorB-datai*step+j*channels+2)*(backgroundk.colorB-datai*step+j*channels+2); prbxij+=exp(-distance/(2*h*h)/(sizeback*h*h*sqrt(2*pi); for(k=0;ksizetarget;k+) distance=(targetk.colorR-datai*step+j*channels)*(targetk.colorR-datai*step+j*channels)+(targetk.colorG-datai*step+j*channels+1)*(targetk.colorG-datai*step+j*channels+1)+(targetk.colorB-datai*step+j*channels+2)*(targetk.colorB-datai*step+j*channels+2); prfxij+=exp(-distance/(2*h*h)/(sizetarget*h*h*sqrt(2*pi); double sumprb,sumprf; sumprb=0.; sumprf=0.; for(i=0;iheight;i+) for(j=0;jwidth;j+) sumprb+=prbxij; sumprf+=prfxij; for(i=0;iheight;i+) for(j=0;jwidth;j+) prbxij/=sumprb; prfxij/=sumprf; for(i=0;iheight;i+) for(j=0;j=prfxij)/这个点属于前景 showdatai*step+j*channels=255; showdatai*step+j*channels+1=255; showdatai*step+j*channels+2=255; cvNamedWindow(showImg,1); cvShowImage(showImg,showImg); cvWaitKey(0); cvDestroyWindow(Interac