一起学opencv (八) 分水岭和选取segment

头文件

#if !defined WATERSHS#define WATERSHS#include <opencv2/core/core.hpp>#include <opencv2/imgproc/imgproc.hpp>class WatershedSegmenter {private:cv::Mat markers;public:void setMarkers(const cv::Mat& markerImage) {// Convert to image of intsmarkerImage.convertTo(markers, CV_32S);}cv::Mat process(const cv::Mat &image) {// Apply watershedcv::watershed(image, markers);return markers;}// Return result in the form of an imagecv::Mat getSegmentation() {cv::Mat tmp;// all segment with label higher than 255// will be assigned value 255markers.convertTo(tmp, CV_8U);return tmp;}// Return watershed in the form of an imagecv::Mat getWatersheds() {cv::Mat tmp;markers.convertTo(tmp, CV_8U, 255, 255);return tmp;}};#endif

mian

/*------------------------------------------------------------------------------------------*\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 removedor 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 imagecv::Mat image = cv::imread("group.jpg");if (!image.data)return 0;// Display the imagecv::namedWindow("Original Image");cv::imshow("Original Image", image);// Get the binary mapcv::Mat binary;binary = cv::imread("binary.bmp", 0);// Display the binary imagecv::namedWindow("Binary Image");cv::imshow("Binary Image", binary);// Eliminate noise and smaller objectscv::Mat fg;cv::erode(binary, fg, cv::Mat(), cv::Point(-1, -1), 6);// Display the foreground imagecv::namedWindow("Foreground Image");cv::imshow("Foreground Image", fg);// Identify image pixels without objectscv::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 imagecv::namedWindow("Background Image");cv::imshow("Background Image", bg);// Show markers imagecv::Mat markers(binary.size(), CV_8U, cv::Scalar(0));markers = fg + bg;//腐蚀图和膨胀图相加，将标定关键点：牛cv::namedWindow("Markers");cv::imshow("Markers", markers);// Create watershed segmentation objectWatershedSegmenter segmenter;// Set markers and processsegmenter.setMarkers(markers);//转为CV_32S格式segmenter.process(image);// Display segmentation resultcv::namedWindow("Segmentation");cv::imshow("Segmentation", segmenter.getSegmentation());// Display watershedscv::namedWindow("Watersheds");cv::imshow("Watersheds", segmenter.getWatersheds());// Open another imageimage = 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 processsegmenter.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 watershedscv::namedWindow("Watersheds of foreground object");cv::imshow("Watersheds of foreground object", segmenter.getWatersheds());// Open another imageimage = 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 segmentationcv::grabCut(image,    // input imageresult,   // segmentation resultrectangle,// rectangle containing foreground bgModel, fgModel, // models1,        // number of iterationscv::GC_INIT_WITH_RECT); // use rectangle// Get the pixels marked as likely foregroundcv::compare(result, cv::GC_PR_FGD, result, cv::CMP_EQ);// Generate output imagecv::Mat foreground(image.size(), CV_8UC3, cv::Scalar(255, 255, 255));image.copyTo(foreground, result); // bg pixels not copied// draw rectangle on original imagecv::rectangle(image, rectangle, cv::Scalar(255, 255, 255), 1);cv::namedWindow("Image");cv::imshow("Image", image);// display resultcv::namedWindow("Segmented Image");cv::imshow("Segmented Image", foreground);// Open another imageimage = cv::imread("group.jpg");// define bounding rectangle cv::Rect rectangle2(10, 100, 380, 180);cv::Mat bkgModel, fgrModel; // the models (internally used)// GrabCut segmentationcv::grabCut(image,  // input image  CV_8UC3result, // segmentation resultrectangle2, 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 imagecv::rectangle(image, rectangle2, cv::Scalar(255, 255, 255), 1);cv::namedWindow("Image 2");cv::imshow("Image 2", image);// display resultcv::namedWindow("Foreground objects");cv::imshow("Foreground objects", foreground);cv::waitKey();return 0;}