opencv实现分水岭算法

               

//  分水岭算法原理//  IplImage* marker_mask = 0;IplImage* markers = 0;//IplImage* img0 = 0, *img = 0, *img_gray = 0, *wshed = 0;IplImage  *img_gray = 0, *wshed = 0;CvPoint prev_pt = {-1,-1};void on_mouse( int event, int x, int y, int flags, void* param ){ if( !img )  return; 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 < 0 )   prev_pt = pt;  cvLine( marker_mask, prev_pt, pt, cvScalarAll(255), 5, 8, 0 );  cvLine( img, prev_pt, pt, cvScalarAll(255), 5, 8, 0 );  prev_pt = pt;  cvShowImage( "image", img ); }}void CCVMFCView::OnWatershed()//分水岭{ int   flag=0; CvRNG rng = cvRNG(-1); img0 = cvCloneImage( workImg );         //  建立工作位图 cvFlip(img0); cvNamedWindow( "image", 1 ); //  cvNamedWindow( "watershed transform", 1 ); img = cvCloneImage( img0 ); img_gray = cvCloneImage( img0 ); wshed = cvCloneImage( img0 ); marker_mask = cvCreateImage( cvGetSize(img), 8, 1 ); markers = cvCreateImage( cvGetSize(img), IPL_DEPTH_32S, 1 ); cvCvtColor( img, marker_mask, CV_BGR2GRAY ); cvCvtColor( marker_mask, img_gray, CV_GRAY2BGR ); cvZero( marker_mask ); cvZero( wshed ); cvShowImage( "image", img ); //  cvShowImage( "watershed transform", wshed ); cvSetMouseCallback( "image", on_mouse, 0 ); m_ImageType=-3; for(;;) {  int c = cvWaitKey(0);  if( c == 27 ) {   if (!flag) {                    //  未加标记    wshed = cvCloneImage( img0 );   }   break;  }  if( c == 'r' )  {   cvZero( marker_mask );   cvCopy( img0, img );   cvShowImage( "image", img );  }  if( c == 'w' || c == '\r' )  {   CvMemStorage* storage = cvCreateMemStorage(0);   CvSeq* contours = 0;   CvMat* color_tab;   int i, j, comp_count = 0;   //cvSaveImage( "wshed_mask.png", marker_mask );   //marker_mask = cvLoadImage( "wshed_mask.png", 0 );   cvFindContours( marker_mask, storage, &contours, sizeof(CvContour),    CV_RETR_CCOMP, CV_CHAIN_APPROX_SIMPLE );   CvSeq* contourn = contours;   int n;   for (n=0; contourn != 0; contourn = contourn->h_next,n++) {   }                               //  检查边界数   if (n) {                        //  已作标记才进行处理                 cvZero( markers );    for( ; contours != 0; contours = contours->h_next, comp_count++ )    {     cvDrawContours( markers, contours, cvScalarAll(comp_count+1),      cvScalarAll(comp_count+1), -1, -1, 8, cvPoint(0,0) );    }    color_tab = cvCreateMat( 1, comp_count, CV_8UC3 );    for( i = 0; i < comp_count; i++ )    {     uchar* ptr = color_tab->data.ptr + i*3;     ptr[0] = (uchar)(cvRandInt(&rng)%180 + 50);     ptr[1] = (uchar)(cvRandInt(&rng)%180 + 50);     ptr[2] = (uchar)(cvRandInt(&rng)%180 + 50);    }    {     double t = (double)cvGetTickCount();     cvWatershed( img0, markers );  //  分水岭算法处理     t = (double)cvGetTickCount() - t;     //     printf( "exec time = %gms\n", t/(cvGetTickFrequency()*1000.) );    }    // paint the watershed image    for( i = 0; i < markers->height; i++ ) {     for( j = 0; j < markers->width; j++ )     {      int idx = CV_IMAGE_ELEM( markers, int, i, j );      uchar* dst = &CV_IMAGE_ELEM( wshed, uchar, i, j*3 );      if( idx == -1 )       dst[0] = dst[1] = dst[2] = (uchar)255;      else if( idx <= 0 || idx > comp_count )       dst[0] = dst[1] = dst[2] = (uchar)0; // should not get here      else      {       uchar* ptr = color_tab->data.ptr + (idx-1)*3;       dst[0] = ptr[0]; dst[1] = ptr[1]; dst[2] = ptr[2];      }     }    }    cvAddWeighted( wshed, 0.5, img_gray, 0.5, 0, wshed );  //  图像合成    //    cvShowImage( "watershed transform", wshed );    cvReleaseMemStorage( &storage );    cvReleaseMat( &color_tab );   }   else {                          //  未加标记    wshed = cvCloneImage( img0 );   }   cvCopy(wshed,workImg);   cvFlip(workImg);   CClientDC dc(this);   StretchDIBits(dc.m_hDC,         //  刷新主窗口    0,0,workImg->width,workImg->height,    0,0,workImg->width,workImg->height,    workImg->imageData,m_lpBmi,DIB_RGB_COLORS,SRCCOPY);   flag=1;  } } cvDestroyWindow( "image" ); cvReleaseImage(&img0); cvReleaseImage(&img); cvReleaseImage(&img_gray); cvReleaseImage(&marker_mask); cvReleaseImage(&markers); cvFlip(wshed); m_dibFlag=imageReplace(wshed,&workImg); Invalidate();}

           

再分享一下我老师大神的人工智能教程吧。零基础！通俗易懂！风趣幽默！还带黄段子！希望你也加入到我们人工智能的队伍中来！https://blog.csdn.net/jiangjunshow