samples/cpp/tutorial_code/ImgProc/Morphology_1.cpp

腐蝕和膨脹示例程式碼

檢視 相應的教程 以獲取更多詳情

 
#include "opencv2/imgproc.hpp"
#include "opencv2/highgui.hpp"
#include <iostream>
 
using namespace cv;
using namespace std;
 
Mat src, erosion_dst, dilation_dst;
 
int erosion_elem = 0;
int erosion_size = 0;
int dilation_elem = 0;
int dilation_size = 0;
int const max_elem = 3;
int const max_kernel_size = 21;
 
void Erosion( int, void* );
void Dilation( int, void* );
 
 
int main( int argc, char** argv )
{
  CommandLineParser parser( argc, argv, "{@input | LinuxLogo.jpg | input image}" );
src = imread( samples::findFile( parser.get<String>( "@input" ) ), IMREAD_COLOR );
  if( src.empty() )
  {
cout << "無法開啟或找到影像!\n" << endl;
cout << "用法: " << argv[0] << " <輸入影像>" << endl;
    return -1;
  }
 
namedWindow( "腐蝕演示", WINDOW_AUTOSIZE );
namedWindow( "膨脹演示", WINDOW_AUTOSIZE );
moveWindow( "膨脹演示", src.cols, 0 );
 
createTrackbar( "元素:\n 0: 矩形 \n 1: 十字形 \n 2: 橢圓形 \n 3: 菱形", "腐蝕演示",
&erosion_elem, max_elem,
Erosion );
 
createTrackbar( "核心大小:\n 2n +1", "腐蝕演示",
&erosion_size, max_kernel_size,
Erosion );
 
createTrackbar( "元素:\n 0: 矩形 \n 1: 十字形 \n 2: 橢圓形 \n 3: 菱形", "膨脹演示",
&dilation_elem, max_elem,
Dilation );
 
createTrackbar( "核心大小:\n 2n +1", "膨脹演示",
&dilation_size, max_kernel_size,
Dilation );
 
Erosion( 0, 0 );
Dilation( 0, 0 );
 
waitKey(0);
  return 0;
}
 
 
void Erosion( int, void* )
{
  int erosion_type = 0;
  if( erosion_elem == 0 ){ erosion_type = MORPH_RECT; }
  else if( erosion_elem == 1 ){ erosion_type = MORPH_CROSS; }
  else if( erosion_elem == 2) { erosion_type = MORPH_ELLIPSE; }
  else if( erosion_elem == 3) { erosion_type = MORPH_DIAMOND; }
 
  Mat element = getStructuringElement( erosion_type,
                       Size( 2*erosion_size + 1, 2*erosion_size+1 ),
                       Point( erosion_size, erosion_size ) );
 
  erode( src, erosion_dst, element );
  imshow( "腐蝕演示", erosion_dst );
}
 
 
void Dilation( int, void* )
{
  int dilation_type = 0;
  if( dilation_elem == 0 ){ dilation_type = MORPH_RECT; }
  else if( dilation_elem == 1 ){ dilation_type = MORPH_CROSS; }
  else if( dilation_elem == 2) { dilation_type = MORPH_ELLIPSE; }
  else if( dilation_elem == 3) { dilation_type = MORPH_DIAMOND; }
 
  Mat element = getStructuringElement( dilation_type,
                       Size( 2*dilation_size + 1, 2*dilation_size+1 ),
                       Point( dilation_size, dilation_size ) );
 
  dilate( src, dilation_dst, element );
  imshow( "膨脹演示", dilation_dst );
}