自定义均值滤波

#include<opencv2\imgproc\imgproc.hpp>
#include"opencv2\highgui\highgui.hpp"

using namespace std;
using namespace cv;

//自定义均值化函数
void myMean(const Mat &src, Mat &dst)
{
	if (!src.data) return;
	for(int i=1;i<src.rows;++i)
		for (int j = 1; j < src.cols; ++j)
		{
			if ((i - 1) >= 0 && (j - 1) >= 0 && (i + 1) < src.rows && (j + 1) < src.cols)
			{
				dst.at<Vec3b>(i, j)[0] =( src.at<Vec3b>(i, j)[0] + src.at<Vec3b>(i - 1, j - 1)[0] + 
					src.at<Vec3b>(i - 1, j)[0] + src.at<Vec3b>(i - 1, j + 1)[0]+ src.at<Vec3b>(i, j - 1)[0] + 
					src.at<Vec3b>(i, j + 1)[0] + src.at<Vec3b>(i + 1, j - 1)[0] + src.at<Vec3b>(i + 1, j)[0] + src.at<Vec3b>(i + 1,j + 1)[0])/9;

				dst.at<Vec3b>(i, j)[1] = (src.at<Vec3b>(i, j)[1] + src.at<Vec3b>(i - 1, j - 1)[1] +
					src.at<Vec3b>(i - 1, j)[1] + src.at<Vec3b>(i - 1, j + 1)[1] + src.at<Vec3b>(i, j - 1)[1] +
					src.at<Vec3b>(i, j + 1)[1] + src.at<Vec3b>(i + 1, j - 1)[1] + src.at<Vec3b>(i + 1, j)[1] + src.at<Vec3b>(i + 1, j + 1)[1]) / 9;

				dst.at<Vec3b>(i, j)[2] = (src.at<Vec3b>(i, j)[2] + src.at<Vec3b>(i - 1, j - 1)[2] +
					src.at<Vec3b>(i - 1, j)[2] + src.at<Vec3b>(i - 1, j + 1)[2] + src.at<Vec3b>(i, j - 1)[2] +
					src.at<Vec3b>(i, j + 1)[2] + src.at<Vec3b>(i + 1, j - 1)[2] + src.at<Vec3b>(i + 1, j)[2] + src.at<Vec3b>(i + 1, j + 1)[2]) / 9;
			}

			else
			{
				dst.at<Vec3b>(i, j)[0] = src.at<Vec3b>(i, j)[0];
				dst.at<Vec3b>(i, j)[1] = src.at<Vec3b>(i, j)[1];
				dst.at<Vec3b>(i, j)[2] = src.at<Vec3b>(i, j)[2];
			}
		}
}
//椒盐化函数
void salt(Mat &image, int n) 
{
	if (!image.data) return;//防止传入空图
	for (int k = 0; k < n; k++)
	{
		int i = rand() % image.rows;
		int j = rand() % image.cols;
		
		if (image.channels() == 1)
		{
			image.at<uchar>(i, j) = 255;
		}
		else if (image.channels() == 3)
		{
			image.at<Vec3b>(i, j) = 255;
		}
	}
}

void main()
{
	Mat image = imread("D:/2015project/meanlvbo/1.jpg");
	Mat Salt_image;
	image.copyTo(Salt_image);
	salt(Salt_image, 3000);

	Mat image1(image.size(), image.type());
	Mat image2;
	myMean(Salt_image, image1);
	blur(Salt_image, image2, Size(3, 3));
	imshow("原图", image);
	imshow("自定义均值滤波", image1);
	imshow("opencv自带均值滤波", image2);
	waitKey(0);
}