Blame view

3rdparty/opencv-4.5.4/samples/cpp/edge.cpp 2.21 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
  #include "opencv2/core/utility.hpp"
  #include "opencv2/imgproc.hpp"
  #include "opencv2/imgcodecs.hpp"
  #include "opencv2/highgui.hpp"
  
  #include <stdio.h>
  
  using namespace cv;
  using namespace std;
  
  int edgeThresh = 1;
  int edgeThreshScharr=1;
  
  Mat image, gray, blurImage, edge1, edge2, cedge;
  
  const char* window_name1 = "Edge map : Canny default (Sobel gradient)";
  const char* window_name2 = "Edge map : Canny with custom gradient (Scharr)";
  
  // define a trackbar callback
  static void onTrackbar(int, void*)
  {
      blur(gray, blurImage, Size(3,3));
  
      // Run the edge detector on grayscale
      Canny(blurImage, edge1, edgeThresh, edgeThresh*3, 3);
      cedge = Scalar::all(0);
  
      image.copyTo(cedge, edge1);
      imshow(window_name1, cedge);
  
      /// Canny detector with scharr
      Mat dx,dy;
      Scharr(blurImage,dx,CV_16S,1,0);
      Scharr(blurImage,dy,CV_16S,0,1);
      Canny( dx,dy, edge2, edgeThreshScharr, edgeThreshScharr*3 );
      /// Using Canny's output as a mask, we display our result
      cedge = Scalar::all(0);
      image.copyTo(cedge, edge2);
      imshow(window_name2, cedge);
  }
  
  static void help(const char** argv)
  {
      printf("\nThis sample demonstrates Canny edge detection\n"
             "Call:\n"
             "    %s [image_name -- Default is fruits.jpg]\n\n", argv[0]);
  }
  
  const char* keys =
  {
      "{help h||}{@image |fruits.jpg|input image name}"
  };
  
  int main( int argc, const char** argv )
  {
      help(argv);
      CommandLineParser parser(argc, argv, keys);
      string filename = parser.get<string>(0);
  
      image = imread(samples::findFile(filename), IMREAD_COLOR);
      if(image.empty())
      {
          printf("Cannot read image file: %s\n", filename.c_str());
          help(argv);
          return -1;
      }
      cedge.create(image.size(), image.type());
      cvtColor(image, gray, COLOR_BGR2GRAY);
  
      // Create a window
      namedWindow(window_name1, 1);
      namedWindow(window_name2, 1);
  
      // create a toolbar
      createTrackbar("Canny threshold default", window_name1, &edgeThresh, 100, onTrackbar);
      createTrackbar("Canny threshold Scharr", window_name2, &edgeThreshScharr, 400, onTrackbar);
  
      // Show the image
      onTrackbar(0, 0);
  
      // Wait for a key stroke; the same function arranges events processing
      waitKey(0);
  
      return 0;
  }