Subversion Repositories OpenCV2-Cookbook

/*------------------------------------------------------------------------------------------*\
   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("../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("../binary.bmp",0);

    // 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::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);

        // Create watershed segmentation object
        WatershedSegmenter segmenter;

        // Set markers and process
        segmenter.setMarkers(markers);
        segmenter.process(image);

        // Display segmentation result
        cv::namedWindow("Segmentation");
        cv::imshow("Segmentation",segmenter.getSegmentation());

        // Display watersheds
        cv::namedWindow("Watersheds");
        cv::imshow("Watersheds",segmenter.getWatersheds());

        // Open another image
        image= cv::imread("../tower.jpg");

        // Identify background pixels
        cv::Mat imageMask(image.size(),CV_8U,cv::Scalar(0));
        cv::rectangle(imageMask,cv::Point(5,5),cv::Point(image.cols-5,image.rows-5),cv::Scalar(255),3);
        // 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);

        // Set markers and process
        segmenter.setMarkers(imageMask);
        segmenter.process(image);

    // 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.getWatersheds());

        // Open another image
        image= cv::imread("../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

        // Get the pixels marked as likely foreground
        cv::compare(result,cv::GC_PR_FGD,result,cv::CMP_EQ);
        // Generate output image
        cv::Mat foreground(image.size(),CV_8UC3,cv::Scalar(255,255,255));
        image.copyTo(foreground,result); // bg pixels not copied

        // 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("../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;
}