1 #include "opencv2/videoio/videoio.hpp"
2 #include "opencv2/highgui/highgui.hpp"
3 #include "opencv2/imgproc/imgproc.hpp"
4
5 #include <ctype.h>
6 #include <stdio.h>
7 #include <iostream>
8
9 using namespace cv;
10 using namespace std;
11
help()12 static void help()
13 {
14 cout <<
15 "\nThis program demonstrates Laplace point/edge detection using OpenCV function Laplacian()\n"
16 "It captures from the camera of your choice: 0, 1, ... default 0\n"
17 "Call:\n"
18 "./laplace [camera #, default 0]\n" << endl;
19 }
20
21 enum {GAUSSIAN, BLUR, MEDIAN};
22
23 int sigma = 3;
24 int smoothType = GAUSSIAN;
25
main(int argc,char ** argv)26 int main( int argc, char** argv )
27 {
28 VideoCapture cap;
29 help();
30
31 if( argc == 1 || (argc == 2 && strlen(argv[1]) == 1 && isdigit(argv[1][0])))
32 cap.open(argc == 2 ? argv[1][0] - '0' : 0);
33 else if( argc >= 2 )
34 {
35 cap.open(argv[1]);
36 if( cap.isOpened() )
37 cout << "Video " << argv[1] <<
38 ": width=" << cap.get(CAP_PROP_FRAME_WIDTH) <<
39 ", height=" << cap.get(CAP_PROP_FRAME_HEIGHT) <<
40 ", nframes=" << cap.get(CAP_PROP_FRAME_COUNT) << endl;
41 if( argc > 2 && isdigit(argv[2][0]) )
42 {
43 int pos;
44 sscanf(argv[2], "%d", &pos);
45 cout << "seeking to frame #" << pos << endl;
46 cap.set(CAP_PROP_POS_FRAMES, pos);
47 }
48 }
49
50 if( !cap.isOpened() )
51 {
52 cout << "Could not initialize capturing...\n";
53 return -1;
54 }
55
56 namedWindow( "Laplacian", 0 );
57 createTrackbar( "Sigma", "Laplacian", &sigma, 15, 0 );
58
59 Mat smoothed, laplace, result;
60
61 for(;;)
62 {
63 Mat frame;
64 cap >> frame;
65 if( frame.empty() )
66 break;
67
68 int ksize = (sigma*5)|1;
69 if(smoothType == GAUSSIAN)
70 GaussianBlur(frame, smoothed, Size(ksize, ksize), sigma, sigma);
71 else if(smoothType == BLUR)
72 blur(frame, smoothed, Size(ksize, ksize));
73 else
74 medianBlur(frame, smoothed, ksize);
75
76 Laplacian(smoothed, laplace, CV_16S, 5);
77 convertScaleAbs(laplace, result, (sigma+1)*0.25);
78 imshow("Laplacian", result);
79
80 int c = waitKey(30);
81 if( c == ' ' )
82 smoothType = smoothType == GAUSSIAN ? BLUR : smoothType == BLUR ? MEDIAN : GAUSSIAN;
83 if( c == 'q' || c == 'Q' || (c & 255) == 27 )
84 break;
85 }
86
87 return 0;
88 }
89