opencv watershed and GrabCut

转自：opencv2

/*------------------------------------------------------------------------------------------*\

   This file contains material supporting chapter 5 of the cookbook:  
   Computer Vision Programming using the OpenCV Library. 
   by Robert Laganiere, Packt Publishing, 2011.


   This program is free software; permission is hereby granted to use, copy, modify, 
   and distribute this source code, or portions thereof, for any purpose, without fee, 
   subject to the restriction that the copyright notice may not be removed 
   or altered from any source or altered source distribution. 
   The software is released on an as-is basis and without any warranties of any kind. 
   In particular, the software is not guaranteed to be fault-tolerant or free from failure. 
   The author disclaims all warranties with regard to this software, any use, 
   and any consequent failure, is purely the responsibility of the user.
 
   Copyright (C) 2010-2011 Robert Laganiere, www.laganiere.name
\*------------------------------------------------------------------------------------------*/


#include <iostream>
#include <opencv2/core/core.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <opencv2/highgui/highgui.hpp>
#include "watershedSegmentation.h"




int main()
{
// Read input image
cv::Mat image= cv::imread(".\\images\\group.jpg");
if (!image.data)
return 0; 


    // Display the image
cv::namedWindow("Original Image");
cv::imshow("Original Image",image);


// Get the binary map
cv::Mat binary;
binary= cv::imread(".\\images\\binary.bmp",0);
std::cout<<binary(cv::Rect(100,100,10,10))<<std::endl;
    // Display the binary image
cv::namedWindow("Binary Image");
cv::imshow("Binary Image",binary);


// Eliminate noise and smaller objects
cv::Mat fg;
cv::erode(binary,fg,cv::Mat(),cv::Point(-1,-1),6);


    // Display the foreground image
cv::namedWindow("Foreground Image");
cv::imshow("Foreground Image",fg);


// Identify image pixels without objects
cv::Mat bg;
cv::dilate(binary,bg,cv::Mat(),cv::Point(-1,-1),6);
cv::namedWindow("dilate");
cv::imshow("dilate",bg);
cv::threshold(bg,bg,1,128,cv::THRESH_BINARY_INV);


    // Display the background image
cv::namedWindow("Background Image");
cv::imshow("Background Image",bg);


// Show markers image
cv::Mat markers(binary.size(),CV_8U,cv::Scalar(0));
markers= fg+bg;
cv::namedWindow("Markers");
cv::imshow("Markers",markers);
std::cout<<"腐蚀以后："<<markers(cv::Rect(1,149,450,1))<<"腐蚀结束"<<std::endl;
// Create watershed segmentation object
WatershedSegmenter segmenter;


// Set markers and process
segmenter.setMarkers(markers);
cv::namedWindow("yuanshi");
cv::imshow("yuanshi",segmenter.process(image));
std::cout<<"废水以后"<<segmenter.process(image)(cv::Rect(1,149,450,1))<<"废水结束"<<std::endl;
// Display segmentation result
cv::namedWindow("Segmentation");
cv::Mat seg;
cv::imshow("Segmentation",segmenter.getSegmentation());
seg=segmenter.getSegmentation();
cv::line(seg,cv::Point(1,149),cv::Point(451,150),cv::Scalar(0));
cv::imshow("Segmentation",seg);


std::cout<<"seg"<<segmenter.getSegmentation()(cv::Rect(1,149,450,1))<<"seg"<<std::endl;
// Display watersheds
cv::namedWindow("Watersheds");
cv::imshow("Watersheds",segmenter.getWatersheds());
std::cout<<"seg2"<<segmenter.getWatersheds()(cv::Rect(1,149,450,1))<<"seg2"<<std::endl;
// Open another image
image= cv::imread(".\\images\\tower.jpg");


// Identify background pixels
cv::Mat imageMask(image.size(),CV_8U,cv::Scalar(0));
cv::namedWindow("mask1");cv::imshow("mask1",imageMask);
cv::rectangle(imageMask,cv::Point(5,5),cv::Point(image.cols-5,image.rows-5),cv::Scalar(255),3);
cv::namedWindow("mask2");cv::imshow("mask2",imageMask);


// Identify foreground pixels (in the middle of the image)
cv::rectangle(imageMask,cv::Point(image.cols/2-10,image.rows/2-10),
cv::Point(image.cols/2+10,image.rows/2+10),cv::Scalar(1),10);
cv::namedWindow("mask3");cv::imshow("mask3",imageMask);


// Set markers and process
segmenter.setMarkers(imageMask);
// cv::namedWindow("mask4");cv::imshow("mask4",segmenter.markers);
cv::Mat maskmask=segmenter.process(image);
cv::namedWindow("mask4");cv::imshow("mask4",maskmask);


std::cout<<maskmask(cv::Rect(240,180,20,20))<<std::endl;


std::cout<<maskmask(cv::Rect(450,1,20,20))<<std::endl;
    
std::cout<<"gl"<<maskmask(cv::Rect(1,180,400,2))<<"glend"<<std::endl;
std::cout<<"以上是maskmask的中间400个和右上角400个，以及中间两行"<<std::endl;
// Display the image with markers
cv::rectangle(image,cv::Point(5,5),cv::Point(image.cols-5,image.rows-5),cv::Scalar(255,255,255),3);
cv::rectangle(image,cv::Point(image.cols/2-10,image.rows/2-10),
cv::Point(image.cols/2+10,image.rows/2+10),cv::Scalar(1,1,1),10);
cv::namedWindow("Image with marker");
cv::imshow("Image with marker",image);


// Display watersheds
cv::namedWindow("Watersheds of foreground object");
cv::imshow("Watersheds of foreground object",segmenter.getSegmentation());
std::cout<<segmenter.getSegmentation()(cv::Rect(240,180,20,20))<<std::endl;


std::cout<<segmenter.getSegmentation()(cv::Rect(450,1,20,20))<<std::endl;
std::cout<<"guoang"<<segmenter.getSegmentation()(cv::Rect(1,180,400,2));
// Open another image
 image= cv::imread(".\\images\\tower.jpg");
// define bounding rectangle 
cv::Rect rectangle(50,70,image.cols-150,image.rows-180);


cv::Mat result; // segmentation result (4 possible values)
cv::Mat bgModel,fgModel; // the models (internally used)
// GrabCut segmentation
cv::grabCut(image,    // input image
       result,   // segmentation result
rectangle,// rectangle containing foreground 
bgModel,fgModel, // models
1,        // number of iterations
cv::GC_INIT_WITH_RECT); // use rectangle
std::cout<<"最开始得到的result的中间一行："<<std::endl<<result(cv::Rect(0,180,470,1))<<"end"<<std::endl;
cv::namedWindow("result1"),cv::imshow("result1",result);
// Get the pixels marked as likely foreground
cv::compare(result,cv::GC_PR_FGD,result,cv::CMP_EQ);

cv::namedWindow("result2"),cv::imshow("result2",result);
std::cout<<"result compare的中间一行："<<std::endl<<result(cv::Rect(0,180,470,1))<<"end"<<std::endl;


// Generate output image
cv::Mat foreground(image.size(),CV_8UC3,cv::Scalar(255,255,255));
cv::namedWindow("result3"),cv::imshow("result3",foreground);


image.copyTo(foreground,result); // bg pixels not copied
cv::namedWindow("result4"),cv::imshow("result4",bgModel);
cv::namedWindow("result5"),cv::imshow("result5",fgModel);


// draw rectangle on original image
cv::rectangle(image, rectangle, cv::Scalar(255,255,255),1);
cv::namedWindow("Image");
cv::imshow("Image",image);


// display result
cv::namedWindow("Segmented Image");
cv::imshow("Segmented Image",foreground);


// Open another image
image= cv::imread(".\\images\\group.jpg");


// define bounding rectangle 
cv::Rect rectangle2(10,100,380,180);


cv::Mat bkgModel,fgrModel; // the models (internally used)
// GrabCut segmentation
cv::grabCut(image,  // input image
       result, // segmentation result
rectangle2,bkgModel,fgrModel,5,cv::GC_INIT_WITH_RECT);
// Get the pixels marked as likely foreground
// cv::compare(result,cv::GC_PR_FGD,result,cv::CMP_EQ);
result= result&1;
foreground.create(image.size(),CV_8UC3);
    foreground.setTo(cv::Scalar(255,255,255));
image.copyTo(foreground,result); // bg pixels not copied


// draw rectangle on original image
cv::rectangle(image, rectangle2, cv::Scalar(255,255,255),1);
cv::namedWindow("Image 2");
cv::imshow("Image 2",image);


// display result
cv::namedWindow("Foreground objects");
cv::imshow("Foreground objects",foreground);


cv::waitKey();
return 0;
}