|
|
|
@ -1,6 +1,6 @@ |
|
|
|
|
/**
|
|
|
|
|
* @file Sobel_Demo.cpp |
|
|
|
|
* @brief Sample code using Sobel and/or Scharr OpenCV functions to make a simple Edge Detector |
|
|
|
|
* @brief Sample code uses Sobel or Scharr OpenCV functions for edge detection |
|
|
|
|
* @author OpenCV team |
|
|
|
|
*/ |
|
|
|
|
|
|
|
|
@ -8,70 +8,110 @@ |
|
|
|
|
#include "opencv2/imgcodecs.hpp" |
|
|
|
|
#include "opencv2/highgui.hpp" |
|
|
|
|
|
|
|
|
|
#include <iostream> |
|
|
|
|
|
|
|
|
|
using namespace cv; |
|
|
|
|
using namespace std; |
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* @function main |
|
|
|
|
*/ |
|
|
|
|
int main( int argc, char** argv ) |
|
|
|
|
{ |
|
|
|
|
cv::CommandLineParser parser(argc, argv, |
|
|
|
|
"{@input |../data/lena.jpg|input image}" |
|
|
|
|
"{ksize k|1|ksize (hit 'K' to increase its value)}" |
|
|
|
|
"{scale s|1|scale (hit 'S' to increase its value)}" |
|
|
|
|
"{delta d|0|delta (hit 'D' to increase its value)}" |
|
|
|
|
"{help h|false|show help message}"); |
|
|
|
|
|
|
|
|
|
cout << "The sample uses Sobel or Scharr OpenCV functions for edge detection\n\n"; |
|
|
|
|
parser.printMessage(); |
|
|
|
|
cout << "\nPress 'ESC' to exit program.\nPress 'R' to reset values ( ksize will be -1 equal to Scharr function )"; |
|
|
|
|
|
|
|
|
|
//![variables]
|
|
|
|
|
Mat src, src_gray; |
|
|
|
|
Mat image,src, src_gray; |
|
|
|
|
Mat grad; |
|
|
|
|
const char* window_name = "Sobel Demo - Simple Edge Detector"; |
|
|
|
|
int scale = 1; |
|
|
|
|
int delta = 0; |
|
|
|
|
const String window_name = "Sobel Demo - Simple Edge Detector"; |
|
|
|
|
int ksize = parser.get<int>("ksize"); |
|
|
|
|
int scale = parser.get<int>("scale"); |
|
|
|
|
int delta = parser.get<int>("delta"); |
|
|
|
|
int ddepth = CV_16S; |
|
|
|
|
//![variables]
|
|
|
|
|
|
|
|
|
|
//![load]
|
|
|
|
|
String imageName("../data/lena.jpg"); // by default
|
|
|
|
|
if (argc > 1) |
|
|
|
|
String imageName = parser.get<String>("@input"); // by default
|
|
|
|
|
image = imread( imageName, IMREAD_COLOR ); // Load an image
|
|
|
|
|
|
|
|
|
|
if( image.empty() ) |
|
|
|
|
{ |
|
|
|
|
imageName = argv[1]; |
|
|
|
|
return 1; |
|
|
|
|
} |
|
|
|
|
src = imread( imageName, IMREAD_COLOR ); // Load an image
|
|
|
|
|
|
|
|
|
|
if( src.empty() ) |
|
|
|
|
{ return -1; } |
|
|
|
|
//![load]
|
|
|
|
|
|
|
|
|
|
//![reduce_noise]
|
|
|
|
|
GaussianBlur( src, src, Size(3,3), 0, 0, BORDER_DEFAULT ); |
|
|
|
|
//![reduce_noise]
|
|
|
|
|
for (;;) |
|
|
|
|
{ |
|
|
|
|
//![reduce_noise]
|
|
|
|
|
GaussianBlur(image, src, Size(3, 3), 0, 0, BORDER_DEFAULT); |
|
|
|
|
//![reduce_noise]
|
|
|
|
|
|
|
|
|
|
//![convert_to_gray]
|
|
|
|
|
cvtColor( src, src_gray, COLOR_BGR2GRAY ); |
|
|
|
|
//![convert_to_gray]
|
|
|
|
|
//![convert_to_gray]
|
|
|
|
|
cvtColor(src, src_gray, COLOR_BGR2GRAY); |
|
|
|
|
//![convert_to_gray]
|
|
|
|
|
|
|
|
|
|
//![sobel]
|
|
|
|
|
/// Generate grad_x and grad_y
|
|
|
|
|
Mat grad_x, grad_y; |
|
|
|
|
Mat abs_grad_x, abs_grad_y; |
|
|
|
|
//![sobel]
|
|
|
|
|
/// Generate grad_x and grad_y
|
|
|
|
|
Mat grad_x, grad_y; |
|
|
|
|
Mat abs_grad_x, abs_grad_y; |
|
|
|
|
|
|
|
|
|
/// Gradient X
|
|
|
|
|
//Scharr( src_gray, grad_x, ddepth, 1, 0, scale, delta, BORDER_DEFAULT );
|
|
|
|
|
Sobel( src_gray, grad_x, ddepth, 1, 0, 3, scale, delta, BORDER_DEFAULT ); |
|
|
|
|
/// Gradient X
|
|
|
|
|
Sobel(src_gray, grad_x, ddepth, 1, 0, ksize, scale, delta, BORDER_DEFAULT); |
|
|
|
|
|
|
|
|
|
/// Gradient Y
|
|
|
|
|
//Scharr( src_gray, grad_y, ddepth, 0, 1, scale, delta, BORDER_DEFAULT );
|
|
|
|
|
Sobel( src_gray, grad_y, ddepth, 0, 1, 3, scale, delta, BORDER_DEFAULT ); |
|
|
|
|
//![sobel]
|
|
|
|
|
/// Gradient Y
|
|
|
|
|
Sobel(src_gray, grad_y, ddepth, 0, 1, ksize, scale, delta, BORDER_DEFAULT); |
|
|
|
|
//![sobel]
|
|
|
|
|
|
|
|
|
|
//![convert]
|
|
|
|
|
convertScaleAbs( grad_x, abs_grad_x ); |
|
|
|
|
convertScaleAbs( grad_y, abs_grad_y ); |
|
|
|
|
//![convert]
|
|
|
|
|
//![convert]
|
|
|
|
|
convertScaleAbs(grad_x, abs_grad_x); |
|
|
|
|
convertScaleAbs(grad_y, abs_grad_y); |
|
|
|
|
//![convert]
|
|
|
|
|
|
|
|
|
|
//![blend]
|
|
|
|
|
/// Total Gradient (approximate)
|
|
|
|
|
addWeighted( abs_grad_x, 0.5, abs_grad_y, 0.5, 0, grad ); |
|
|
|
|
//![blend]
|
|
|
|
|
//![blend]
|
|
|
|
|
/// Total Gradient (approximate)
|
|
|
|
|
addWeighted(abs_grad_x, 0.5, abs_grad_y, 0.5, 0, grad); |
|
|
|
|
//![blend]
|
|
|
|
|
|
|
|
|
|
//![display]
|
|
|
|
|
imshow( window_name, grad ); |
|
|
|
|
waitKey(0); |
|
|
|
|
//![display]
|
|
|
|
|
//![display]
|
|
|
|
|
imshow(window_name, grad); |
|
|
|
|
char key = (char)waitKey(0); |
|
|
|
|
//![display]
|
|
|
|
|
|
|
|
|
|
if(key == 27) |
|
|
|
|
{ |
|
|
|
|
return 0; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if (key == 'k' || key == 'K') |
|
|
|
|
{ |
|
|
|
|
ksize = ksize < 30 ? ksize+2 : -1; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if (key == 's' || key == 'S') |
|
|
|
|
{ |
|
|
|
|
scale++; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if (key == 'd' || key == 'D') |
|
|
|
|
{ |
|
|
|
|
delta++; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if (key == 'r' || key == 'R') |
|
|
|
|
{ |
|
|
|
|
scale = 1; |
|
|
|
|
ksize = -1; |
|
|
|
|
delta = 0; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
return 0; |
|
|
|
|
} |
|
|
|
|