1 // meanshift_segmentation.cpp : 定义控制台应用程序的入口点。
 2 //
 3
 4 #include "stdafx.h"
 5 #include <opencv2/core/core.hpp>
 6 #include <opencv2/highgui/highgui.hpp>
 7 #include <opencv2/imgproc/imgproc.hpp>
 8 #include <iostream>
 9
10 using namespace cv;
11 using namespace std;
12
13
14 Mat src,dst;
15 int spatialRad=10,colorRad=10,maxPryLevel=1;
16 //const Scalar& colorDiff=Scalar::all(1);
17
18 void meanshift_seg(int,void *)
19 {
20     //调用meanshift图像金字塔进行分割
21     pyrMeanShiftFiltering(src,dst,spatialRad,colorRad,maxPryLevel);
22     RNG rng=theRNG();
23     Mat mask(dst.rows+2,dst.cols+2,CV_8UC1,Scalar::all(0));
24     for(int i=0;i<dst.rows;i++)    //opencv图像等矩阵也是基于0索引的
25         for(int j=0;j<dst.cols;j++)
26             if(mask.at<uchar>(i+1,j+1)==0)
27             {
28                 Scalar newcolor(rng(256),rng(256),rng(256));
29                 floodFill(dst,mask,Point(i,j),newcolor,0,Scalar::all(1),Scalar::all(1));
30         //        floodFill(dst,mask,Point(i,j),newcolor,0,colorDiff,colorDiff);
31             }
32     imshow("dst",dst);
33 }
34
35
36 int main(int argc, uchar* argv[])
37 {
38
39     namedWindow("src",WINDOW_AUTOSIZE);
40     namedWindow("dst",WINDOW_AUTOSIZE);
41
42     src=imread("stuff.jpg");
43     CV_Assert(!src.empty());
44
45     spatialRad=10;
46     colorRad=10;
47     maxPryLevel=1;
48
49     //虽然createTrackbar函数的参数onChange函数要求其2个参数形式为onChange(int,void*)
50     //但是这里是系统响应函数，在使用createTrackbar函数时，其调用的函数可以不用写参数，甚至
51     //括号都不用写，但是其调用函数的实现过程中还是需要满足(int,void*)2个参数类型
52     createTrackbar("spatialRad","dst",&spatialRad,80,meanshift_seg);
53     createTrackbar("colorRad","dst",&colorRad,60,meanshift_seg);
54     createTrackbar("maxPryLevel","dst",&maxPryLevel,5,meanshift_seg);
55
56 //    meanshift_seg(0,0);
57
58     imshow("src",src);
59     /*char c=(char)waitKey();
60     if(27==c)
61         return 0;*/
62     imshow("dst",src);
63     waitKey();//无限等待用户交互响应
64 //    while(1);//这里不能用while(1)的原因是需要等待用户的交互，而while(1)没有该功能。虽然2者都有无限等待的作用。
65     return 0;
66 }