samples/cpp/tutorial_code/ImgTrans/Sobel_Demo.cpp

使用 Sobel 和/或 Scharr OpenCV 函数制作简单边缘检测器的示例代码

查看 相应的教程 了解更多详情

 
#include "opencv2/imgproc.hpp"
#include "opencv2/imgcodecs.hpp"
#include "opencv2/highgui.hpp"
 
#include <iostream>
 
using namespace cv;
using namespace std;
 
int main( int argc, char** argv )
{
  cv::CommandLineParser parser(argc, argv,
                               "{@input |lena.jpg|输入图像}"
                               "{ksize k|1|ksize (运行时按 'K' 键增加其值)}"
                               "{scale s|1|比例 (运行时按 '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;
  const String window_name = "Sobel 演示 - 简单边缘检测器";
  int ksize = parser.get<int>("ksize");
  int scale = parser.get<int>("scale");
  int delta = parser.get<int>("delta");
  int ddepth = CV_16S;
 
  String imageName = parser.get<String>("@input");
  // 像往常一样，我们加载源图像 (src)
image = imread( samples::findFile( imageName ), IMREAD_COLOR ); // 加载图像
 
  // 检查图像是否加载成功
  if( image.empty() )
  {
printf("打开图像时出错: %s\n", imageName.c_str());
    return EXIT_FAILURE;
  }
 
  for (;;)
  {
    // 通过使用高斯滤波器进行模糊处理来消除噪声 (内核大小 = 3)
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;
 
Sobel(src_gray, grad_x, ddepth, 1, 0, ksize, scale, delta, BORDER_DEFAULT);
 
Sobel(src_gray, grad_y, ddepth, 0, 1, ksize, scale, delta, BORDER_DEFAULT);
 
    // 转换回 CV_8U
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);
    char key = (char)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;
}