根據各個濾波公式及前幾章內容，可實現理想低通、理想高通、巴特沃思低通、巴特沃思高通、高斯低通、高斯高通濾波器；
效果圖：
高斯低通 D0=30
巴特沃思低通 D0=30,n=2
理想低通D0=30：
高斯高通D0=80：
巴特沃思高通 D0=80,n=2
理想高通D0=80：
代碼實現：
#include<opencv2/opencv.hpp>
#include<iostream>
using namespace std;
using namespace cv;
cv::Mat image_make_border(cv::Mat &src )
{
int w=getOptimalDFTSize(src.cols);
int h=getOptimalDFTSize(src.rows);
Mat padded;
copyMakeBorder(src,padded,0,h-src.rows,0,w-src.cols,BORDER_CONSTANT,Scalar::all(0));
padded.convertTo(padded,CV_32FC1);
return padded;
}
//頻率域濾波
Mat frequency_filter(Mat &scr,Mat &blur)
{
//***********************DFT*******************
Mat plane[]={scr, Mat::zeros(scr.size() , CV_32FC1)}; //創建通道，存儲dft後的實部與虛部（CV_32F，必須爲單通道數）
Mat complexIm;
merge(plane,2,complexIm);//合併通道 （把兩個矩陣合併爲一個2通道的Mat類容器）
dft(complexIm,complexIm);//進行傅立葉變換，結果保存在自身
//***************中心化********************
split(complexIm,plane);//分離通道（數組分離）
// plane[0] = plane[0](Rect(0, 0, plane[0].cols & -2, plane[0].rows & -2));//這裏爲什麼&上-2具體查看opencv文檔
// //其實是爲了把行和列變成偶數 -2的二進制是11111111.......10 最後一位是0
int cx=plane[0].cols/2;int cy=plane[0].rows/2;//以下的操作是移動圖像 (零頻移到中心)
Mat part1_r(plane[0],Rect(0,0,cx,cy)); //元素座標表示爲(cx,cy)
Mat part2_r(plane[0],Rect(cx,0,cx,cy));
Mat part3_r(plane[0],Rect(0,cy,cx,cy));
Mat part4_r(plane[0],Rect(cx,cy,cx,cy));
Mat temp;
part1_r.copyTo(temp); //左上與右下交換位置(實部)
part4_r.copyTo(part1_r);
temp.copyTo(part4_r);
part2_r.copyTo(temp); //右上與左下交換位置(實部)
part3_r.copyTo(part2_r);
temp.copyTo(part3_r);
Mat part1_i(plane[1],Rect(0,0,cx,cy)); //元素座標(cx,cy)
Mat part2_i(plane[1],Rect(cx,0,cx,cy));
Mat part3_i(plane[1],Rect(0,cy,cx,cy));
Mat part4_i(plane[1],Rect(cx,cy,cx,cy));
part1_i.copyTo(temp); //左上與右下交換位置(虛部)
part4_i.copyTo(part1_i);
temp.copyTo(part4_i);
part2_i.copyTo(temp); //右上與左下交換位置(虛部)
part3_i.copyTo(part2_i);
temp.copyTo(part3_i);
//*****************濾波器函數與DFT結果的乘積****************
Mat blur_r,blur_i,BLUR;
multiply(plane[0], blur, blur_r); //濾波（實部與濾波器模板對應元素相乘）
multiply(plane[1], blur,blur_i);//濾波（虛部與濾波器模板對應元素相乘）
Mat plane1[]={blur_r, blur_i};
merge(plane1,2,BLUR);//實部與虛部合併
//*********************得到原圖頻譜圖***********************************
magnitude(plane[0],plane[1],plane[0]);//獲取幅度圖像，0通道爲實部通道，1爲虛部，因爲二維傅立葉變換結果是複數
plane[0]+=Scalar::all(1); //傅立葉變換後的圖片不好分析，進行對數處理，結果比較好看
log(plane[0],plane[0]); // float型的灰度空間爲[0，1])
normalize(plane[0],plane[0],1,0,CV_MINMAX); //歸一化便於顯示
idft( BLUR, BLUR); //idft結果也爲複數
split(BLUR,plane);//分離通道，主要獲取通道
magnitude(plane[0],plane[1],plane[0]); //求幅值(模)
normalize(plane[0],plane[0],1,0,CV_MINMAX); //歸一化便於顯示
return plane[0];//返回參數
}
//*****************理想低通濾波器***********************
Mat ideal_low_kernel(Mat &scr,float sigma)
{
Mat ideal_low_pass(scr.size(),CV_32FC1); //，CV_32FC1
float d0=sigma;//半徑D0越小，模糊越大；半徑D0越大，模糊越小
for(int i=0;i<scr.rows ; i++ ){
for(int j=0; j<scr.cols ; j++ ){
double d = sqrt(pow((i - scr.rows/2),2) + pow((j - scr.cols/2),2));//分子,計算pow必須爲float型
if (d <= d0){
ideal_low_pass.at<float>(i,j)=1;
}else{
ideal_low_pass.at<float>(i,j)=0;
}
}
}
string name = "理想低通濾波器d0=" + std::to_string(sigma);
imshow(name, ideal_low_pass);
return ideal_low_pass;
}
//理想低通濾波器
cv::Mat ideal_low_pass_filter(Mat &src, float sigma)
{
Mat padded = image_make_border(src);
Mat ideal_kernel=ideal_low_kernel(padded,sigma);
Mat result = frequency_filter(padded,ideal_kernel);
return result;
}
Mat butterworth_low_kernel(Mat &scr,float sigma, int n)
{
Mat butterworth_low_pass(scr.size(),CV_32FC1); //，CV_32FC1
double D0 = sigma;//半徑D0越小，模糊越大；半徑D0越大，模糊越小
for(int i=0;i<scr.rows ; i++ ){
for(int j=0; j<scr.cols ; j++ ){
double d = sqrt(pow((i - scr.rows/2),2) + pow((j - scr.cols/2),2));//分子,計算pow必須爲float型
butterworth_low_pass.at<float>(i,j)=1.0 / (1 + pow(d / D0, 2 * n));
}
}
string name = "巴特沃斯低通濾波器d0=" + std::to_string(sigma) + "n=" + std::to_string(n);
imshow(name, butterworth_low_pass);
return butterworth_low_pass;
}
//巴特沃斯低通濾波器
Mat butterworth_low_paass_filter(Mat &src, float d0, int n)
{
//H = 1 / (1+(D/D0)^2n) n表示巴特沃斯濾波器的次數
//階數n=1 無振鈴和負值 階數n=2 輕微振鈴和負值 階數n=5 明顯振鈴和負值 階數n=20 與ILPF相似
Mat padded = image_make_border(src);
Mat butterworth_kernel=butterworth_low_kernel(padded,d0, n);
Mat result = frequency_filter(padded,butterworth_kernel);
return result;
}
Mat gaussian_low_pass_kernel(Mat scr,float sigma)
{
Mat gaussianBlur(scr.size(),CV_32FC1); //，CV_32FC1
float d0=2*sigma*sigma;//高斯函數參數，越小，頻率高斯濾波器越窄，濾除高頻成分越多，圖像就越平滑
for(int i=0;i<scr.rows ; i++ ){
for(int j=0; j<scr.cols ; j++ ){
float d=pow(float(i-scr.rows/2),2)+pow(float(j-scr.cols/2),2);//分子,計算pow必須爲float型
gaussianBlur.at<float>(i,j)=expf(-d/d0);//expf爲以e爲底求冪（必須爲float型）
}
}
// Mat show = gaussianBlur.clone();
// //歸一化到[0,255]供顯示
// normalize(show, show, 0, 255, NORM_MINMAX);
// //轉化成CV_8U型
// show.convertTo(show, CV_8U);
// std::string pic_name = "gaussi" + std::to_string((int)sigma) + ".jpg";
// imwrite( pic_name, show);
imshow("高斯低通濾波器",gaussianBlur);
return gaussianBlur;
}
//高斯低通
Mat gaussian_low_pass_filter(Mat &src, float d0)
{
Mat padded = image_make_border(src);
Mat gaussian_kernel=gaussian_low_pass_kernel(padded,d0);//理想低通濾波器
Mat result = frequency_filter(padded,gaussian_kernel);
return result;
}
Mat ideal_high_kernel(Mat &scr,float sigma)
{
Mat ideal_high_pass(scr.size(),CV_32FC1); //，CV_32FC1
float d0=sigma;//半徑D0越小，模糊越大；半徑D0越大，模糊越小
for(int i=0;i<scr.rows ; i++ ){
for(int j=0; j<scr.cols ; j++ ){
double d = sqrt(pow((i - scr.rows/2),2) + pow((j - scr.cols/2),2));//分子,計算pow必須爲float型
if (d <= d0){
ideal_high_pass.at<float>(i,j)=0;
}
else{
ideal_high_pass.at<float>(i,j)=1;
}
}
}
string name = "理想高通濾波器d0=" + std::to_string(sigma);
imshow(name, ideal_high_pass);
return ideal_high_pass;
}
//理想高通濾波器
cv::Mat ideal_high_pass_filter(Mat &src, float sigma)
{
Mat padded = image_make_border(src);
Mat ideal_kernel=ideal_high_kernel(padded,sigma);
Mat result = frequency_filter(padded,ideal_kernel);
return result;
}
Mat butterworth_high_kernel(Mat &scr,float sigma, int n)
{
Mat butterworth_low_pass(scr.size(),CV_32FC1); //，CV_32FC1
double D0 = sigma;//半徑D0越小，模糊越大；半徑D0越大，模糊越小
for(int i=0;i<scr.rows ; i++ ){
for(int j=0; j<scr.cols ; j++ ){
double d = sqrt(pow((i - scr.rows/2),2) + pow((j - scr.cols/2),2));//分子,計算pow必須爲float型
butterworth_low_pass.at<float>(i,j)=1.0 / (1 + pow(D0 / d, 2 * n));
}
}
string name = "巴特沃斯高通濾波器d0=" + std::to_string(sigma) + "n=" + std::to_string(n);
imshow(name, butterworth_low_pass);
return butterworth_low_pass;
}
//巴特沃斯高通濾波器
Mat butterworth_high_paass_filter(Mat &src, float d0, int n)
{
//H = 1 / (1+(D0/D)^2n) n表示巴特沃斯濾波器的次數
Mat padded = image_make_border(src);
Mat butterworth_kernel=butterworth_high_kernel(padded,d0, n);
Mat result = frequency_filter(padded,butterworth_kernel);
return result;
}
Mat gaussian_high_pass_kernel(Mat scr,float sigma)
{
Mat gaussianBlur(scr.size(),CV_32FC1); //，CV_32FC1
float d0=2*sigma*sigma;
for(int i=0;i<scr.rows ; i++ ){
for(int j=0; j<scr.cols ; j++ ){
float d=pow(float(i-scr.rows/2),2)+pow(float(j-scr.cols/2),2);//分子,計算pow必須爲float型
gaussianBlur.at<float>(i,j)=1-expf(-d/d0);
}
}
imshow("高斯高通濾波器",gaussianBlur);
return gaussianBlur;
}
//高斯高通
Mat gaussian_high_pass_filter(Mat &src, float d0)
{
Mat padded = image_make_border(src);
Mat gaussian_kernel=gaussian_high_pass_kernel(padded,d0);//理想低通濾波器
Mat result = frequency_filter(padded,gaussian_kernel);
return result;
}
int main( int argc, char *argv[])
{
const char* filename = argc >=2 ? argv[1] : "../data/lena.jpg";
Mat input = imread(filename, IMREAD_GRAYSCALE);
if( input.empty())
return -1;
imshow("input",input);//顯示原圖
cv::Mat ideal_low = ideal_low_pass_filter(input, 30);
ideal_low = ideal_low(cv::Rect(0,0, input.cols, input.rows));
imshow("理想低通", ideal_low);
cv::Mat bw_low = butterworth_low_paass_filter(input, 30, 2);
bw_low = bw_low(cv::Rect(0,0, input.cols, input.rows));
imshow("巴特沃斯低通", bw_low);
cv::Mat gaussion_low = gaussian_low_pass_filter(input, 30);
gaussion_low = gaussion_low(cv::Rect(0,0, input.cols, input.rows));
imshow("高斯低通", gaussion_low);
cv::Mat ideal_high = ideal_high_pass_filter(input, 80);
ideal_high = ideal_high(cv::Rect(0,0, input.cols, input.rows));
imshow("理想高通", ideal_high);
cv::Mat bw_high = butterworth_high_paass_filter(input, 80, 2);
bw_high = bw_high(cv::Rect(0,0, input.cols, input.rows));
imshow("巴特沃斯高通", bw_high);
cv::Mat gaussion_high = gaussian_high_pass_filter(input, 80);
gaussion_high = gaussion_high(cv::Rect(0,0, input.cols, input.rows));
imshow("高斯高通", gaussion_high);
waitKey();
return 0;
}