samples/cpp/tutorial_code/ImgProc/Smoothing/Smoothing.cpp

簡單濾波器的示例程式碼

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

 
#include <iostream>
#include "opencv2/imgproc.hpp"
#include "opencv2/imgcodecs.hpp"
#include "opencv2/highgui.hpp"
 
using namespace std;
using namespace cv;
 
int DELAY_CAPTION = 1500;
int DELAY_BLUR = 100;
int MAX_KERNEL_LENGTH = 31;
 
Mat src; Mat dst;
char window_name[] = "平滑演示";
 
int display_caption( const char* caption );
int display_dst( int delay );
 
 
int main( int argc, char ** argv )
{
namedWindow( window_name, WINDOW_AUTOSIZE );
 
    const char* filename = argc >=2 ? argv[1] : "lena.jpg";
 
src = imread( samples::findFile( filename ), IMREAD_COLOR );
    if (src.empty())
    {
printf(" 錯誤開啟影像\n");
printf(" 用法:\n %s [image_name-- 預設為 lena.jpg] \n", argv[0]);
        return EXIT_FAILURE;
    }
 
    if( display_caption( "原始影像" ) != 0 )
    {
        return 0;
    }
 
dst = src.clone();
    if( display_dst( DELAY_CAPTION ) != 0 )
    {
        return 0;
    }
 
    if( display_caption( "均值模糊" ) != 0 )
    {
        return 0;
    }
 
    for ( int i = 1; i < MAX_KERNEL_LENGTH; i = i + 2 )
    {
        blur( src, dst, Size( i, i ), Point(-1,-1) );
        if( display_dst( DELAY_BLUR ) != 0 )
        {
            return 0;
        }
    }
 
    if( display_caption( "高斯模糊" ) != 0 )
    {
        return 0;
    }
 
    for ( int i = 1; i < MAX_KERNEL_LENGTH; i = i + 2 )
    {
        GaussianBlur( src, dst, Size( i, i ), 0, 0 );
        if( display_dst( DELAY_BLUR ) != 0 )
        {
            return 0;
        }
    }
 
    if( display_caption( "中值模糊" ) != 0 )
    {
        return 0;
    }
 
    for ( int i = 1; i < MAX_KERNEL_LENGTH; i = i + 2 )
    {
        medianBlur ( src, dst, i );
        if( display_dst( DELAY_BLUR ) != 0 )
        {
            return 0;
        }
    }
 
    if( display_caption( "雙邊濾波" ) != 0 )
    {
        return 0;
    }
 
    for ( int i = 1; i < MAX_KERNEL_LENGTH; i = i + 2 )
    {
        bilateralFilter ( src, dst, i, i*2, i/2 );
        if( display_dst( DELAY_BLUR ) != 0 )
        {
            return 0;
        }
    }
 
display_caption( "完成!" );
 
    return 0;
}
 
int display_caption( const char* caption )
{
dst = Mat::zeros( src.size(), src.type() );
    putText( dst, caption,
             Point( src.cols/4, src.rows/2),
FONT_HERSHEY_COMPLEX, 1, Scalar(255, 255, 255) );
 
    return display_dst(DELAY_CAPTION);
}
 
int display_dst( int delay )
{
    imshow( window_name, dst );
    int c = waitKey ( delay );
    if( c >= 0 ) { return -1; }
    return 0;
}