借助于opencv的数据结构Mat 实现图像处理中值滤波算法
代码中刻意添加了椒盐噪声,凸显中值滤波的效果,改代码并未考虑边缘影响。
#include <iostream>
#include "algorithm"
#include "opencv2/highgui.hpp"
#include "opencv2/core/core.hpp"
#include "opencv2/opencv.hpp"
int mymedian(int n1,int n2,int n3,int n4,int n5,int n6,int n7,int n8,int n9)
{
int arr[9]={n1,n2,n3,n4,n5,n6,n7,n8,n9};
std::sort(arr,arr+9);
return arr[4];
}
void mymedian_Blur(cv::Mat& src,cv::Mat& dst)
{
cv::Mat img;
src.copyTo(img);
cv::Mat dst_(img.size(),img.type());
int rows=img.rows,cols=img.cols;
for(int r=1;r<rows-1;++r)
{
for(int c=1;c<cols-1;++c)
{
dst_.at<uchar>(r,c)= mymedian(img.at<uchar>(r-1,c-1),img.at<uchar>(r-1,c),img.at<uchar>(r-1,c+1),img.at<uchar>(r,c-1),img.at<uchar>(r,c),img.at<uchar>(r,c+1),img.at<uchar>(r+1,c-1), img.at<uchar>(r+1,c),img.at<uchar>(r+1,c+1));
}
}
dst_.copyTo(dst);
}
void salt(cv::Mat& src,int num)
{
cv::RNG rng(12345);//使用种子seed产生一个64位随机整数
//rng.uniform(a,b);//uniform产生一个[a,b)范围内均匀分布的随机数
//rng.gaussian(a);//gaussian() 产生一个均值为0,标准差为a的随机数(高斯分布)
if(!src.data)
{
return;
}
int rows=src.rows,cols=src.cols;
for(int i=0;i<num;++i)
{
int c=rng.uniform(0,cols);//uniform产生一个均匀分布的随机数
int r=rng.uniform(0,rows);
src.at<cv::Vec3b>(r,c)=cv::Vec3b(255,255,255);
src.at<uchar>(r,c);
}
// cv::imshow("salt",src);
}
int main()
{
std::string imgDir="/home/.../images/anni.jpg";
cv::Mat img=cv::imread(imgDir);
if(img.empty())
{
std::cout<<"file fail to load!"<<std::endl;
return -1;
}
salt(img,10000);
cv::resize(img,img,cv::Size(640,480),0,0,cv::INTER_LINEAR);
cv::imshow("Raw",img);
cv::cvtColor(img,img,cv::COLOR_BGR2GRAY);
int median= mymedian(4,2,5,1,6,7,34,65,23);
std::cout<<"median:"<<median<<std::endl;
cv::Mat dst;
mymedian_Blur(img,dst);
cv::imshow("Test",dst);
cv::waitKey(0);
std::cout << "Hello, World!" << std::endl;
return 0;
}
运行结果:
参考:
https://blog.csdn.net/weixin_37720172/article/details/72627543 中值滤波