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