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|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;
  const String window_name = "Sobel Demo - 简单边缘检测器";
  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;
}