Blame view

3rdparty/opencv-4.5.4/samples/cpp/morphology2.cpp 3.07 KB
f4334277   Hu Chunming   提交3rdparty
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
  #include "opencv2/imgproc.hpp"
  #include "opencv2/imgcodecs.hpp"
  #include "opencv2/highgui.hpp"
  #include <stdlib.h>
  #include <stdio.h>
  #include <string>
  
  using namespace cv;
  
  static void help(char** argv)
  {
  
  printf("\nShow off image morphology: erosion, dialation, open and close\n"
      "Call:\n   %s [image]\n"
      "This program also shows use of rect, ellipse and cross kernels\n\n", argv[0]);
  printf( "Hot keys: \n"
      "\tESC - quit the program\n"
      "\tr - use rectangle structuring element\n"
      "\te - use elliptic structuring element\n"
      "\tc - use cross-shaped structuring element\n"
      "\tSPACE - loop through all the options\n" );
  }
  
  Mat src, dst;
  
  int element_shape = MORPH_RECT;
  
  //the address of variable which receives trackbar position update
  int max_iters = 10;
  int open_close_pos = 0;
  int erode_dilate_pos = 0;
  
  // callback function for open/close trackbar
  static void OpenClose(int, void*)
  {
      int n = open_close_pos;
      int an = abs(n);
      Mat element = getStructuringElement(element_shape, Size(an*2+1, an*2+1), Point(an, an) );
      if( n < 0 )
          morphologyEx(src, dst, MORPH_OPEN, element);
      else
          morphologyEx(src, dst, MORPH_CLOSE, element);
      imshow("Open/Close",dst);
  }
  
  // callback function for erode/dilate trackbar
  static void ErodeDilate(int, void*)
  {
      int n = erode_dilate_pos;
      int an = abs(n);
      Mat element = getStructuringElement(element_shape, Size(an*2+1, an*2+1), Point(an, an) );
      if( n < 0 )
          erode(src, dst, element);
      else
          dilate(src, dst, element);
      imshow("Erode/Dilate",dst);
  }
  
  
  int main( int argc, char** argv )
  {
      cv::CommandLineParser parser(argc, argv, "{help h||}{ @image | baboon.jpg | }");
      if (parser.has("help"))
      {
          help(argv);
          return 0;
      }
      std::string filename = samples::findFile(parser.get<std::string>("@image"));
      if( (src = imread(filename,IMREAD_COLOR)).empty() )
      {
          help(argv);
          return -1;
      }
  
      //create windows for output images
      namedWindow("Open/Close",1);
      namedWindow("Erode/Dilate",1);
  
      open_close_pos = erode_dilate_pos = max_iters;
      createTrackbar("iterations", "Open/Close",&open_close_pos,max_iters*2+1,OpenClose);
      setTrackbarMin("iterations", "Open/Close", -max_iters);
      setTrackbarMax("iterations", "Open/Close", max_iters);
      setTrackbarPos("iterations", "Open/Close", 0);
  
      createTrackbar("iterations", "Erode/Dilate",&erode_dilate_pos,max_iters*2+1,ErodeDilate);
      setTrackbarMin("iterations", "Erode/Dilate", -max_iters);
      setTrackbarMax("iterations", "Erode/Dilate", max_iters);
      setTrackbarPos("iterations", "Erode/Dilate", 0);
  
      for(;;)
      {
          OpenClose(open_close_pos, 0);
          ErodeDilate(erode_dilate_pos, 0);
          char c = (char)waitKey(0);
  
          if( c == 27 )
              break;
          if( c == 'e' )
              element_shape = MORPH_ELLIPSE;
          else if( c == 'r' )
              element_shape = MORPH_RECT;
          else if( c == 'c' )
              element_shape = MORPH_CROSS;
          else if( c == ' ' )
              element_shape = (element_shape + 1) % 3;
      }
  
      return 0;
  }