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;
}