#include <opencv2/opencv.hpp>
#include <iostream>
using namespace cv;
using namespace std;
int main(int argc, char** argv)
{
CommandLineParser parser(argc, argv,
"{@input |lena.jpg|输入图像}"
"{ksize k|1|ksize(在运行时按'K'以增加其值)}"
"{scale s|1|scale(在运行时按'S'以增加其值)}"
"{delta d|0|delta(在运行时按'D'以增加其值)}"
"{help h|false|显示帮助消息}");
cout << "该示例使用 Sobel 或 Scharr OpenCV 函数进行边缘检测\n\n";
parser.printMessage();
cout << "\n按'ESC'退出程序。\n按'R'重置值(ksize 将为 -1,等于 Scharr 函数)";
Mat image, src, src_gray;
Mat grad_x, grad_y;
Mat abs_grad_x, abs_grad_y;
Mat grad;
const string window_name = "Sobel 演示 - 简单边缘检测器";
int ksize = parser.get<int>("ksize");
double scale = parser.get<double>("scale");
double delta = parser.get<double>("delta");
int ddepth = CV_16S;
string imageName = parser.get<string>("@input");
image = imread(imageName, IMREAD_COLOR);
if (image.empty())
{
printf("打开图像时出错:%s\n", imageName.c_str());
return EXIT_FAILURE;
}
for (;;)
{
GaussianBlur(image, src, Size(3, 3), 0, 0, BORDER_DEFAULT);
cvtColor(src, src_gray, COLOR_BGR2GRAY);
Mat grad_x, grad_y;
Mat abs_grad_x, abs_grad_y;
int kernelSize = (ksize == -1) ? -1 : ksize;
Sobel(src_gray, grad_x, ddepth, 1, 0, kernelSize, scale, delta, BORDER_DEFAULT);
Sobel(src_gray, grad_y, ddepth, 0, 1, kernelSize, scale, delta, BORDER_DEFAULT);
convertScaleAbs(grad_x, abs_grad_x);
convertScaleAbs(grad_y, abs_grad_y);
addWeighted(abs_grad_x, 0.5, abs_grad_y, 0.5, 0, grad);
imshow(window_name, grad);
int key = waitKey(0);
if (key == 27)
{
return EXIT_SUCCESS;
}
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 EXIT_SUCCESS;
}