中值滤波

最新推荐文章于 2024-09-08 00:15:00 发布

singing1001

最新推荐文章于 2024-09-08 00:15:00 发布

阅读量188

点赞数

分类专栏：图像处理

本文链接：https://blog.csdn.net/qingsong1001/article/details/109176189

版权

图像处理专栏收录该内容

9 篇文章 0 订阅

订阅专栏

#include <stdio.h>
#include <float.h>
#include <cstdlib>
#include <string>

#include <opencv2/highgui.hpp>
#include "opencv2/imgproc.hpp"
#include <iostream>
#include <fstream>

//using 5x5 window
#define FILTER_WINDOW_Y 5
#define FILTER_WINDOW_X 5
#define FILTER_WINDOW_SIZE 25
#define FILTER_WINDOW_MEDIAN_SIZE (25/2)
#define PIXEL_RANGE_MAX 256

void median_filter( cv::Mat &src, size_t row, unsigned char *pdst)
{
unsigned char *pcurdst = pdst+FILTER_WINDOW_X/2;
unsigned char histogram[PIXEL_RANGE_MAX];
memset(histogram, 0, sizeof(histogram));

for(int i = row-FILTER_WINDOW_Y/2; i <= row+FILTER_WINDOW_Y/2; i++)
{
unsigned char *pRow = src.ptr<unsigned char>(i);
for(int j = 0; j < FILTER_WINDOW_X; j++)
{
histogram[*(pRow+j)] +=1;
}
}
int index = 0;
int count = 0;
for(index = 0, count = 0; index < PIXEL_RANGE_MAX;)
{
count+= histogram[index];
++index;
if(count > FILTER_WINDOW_MEDIAN_SIZE)
{
break;
}
}
*pcurdst = index;
++pcurdst;
//printf("\n %s:%d, index %d, count %d\n", __FILE__, __LINE__, index, count);

for(int col = 1; col < src.cols-FILTER_WINDOW_Y+1; col++, pcurdst++)
{
for(int i = row-FILTER_WINDOW_Y/2; i <= row+FILTER_WINDOW_Y/2; i++)
{
unsigned char *pRow = src.ptr<unsigned char>(i);
histogram[*(pRow+col-1)] -=1;
histogram[*(pRow+col+4)] +=1;
if(*(pRow+col-1) < index)
--count;
if(*(pRow+col+4) < index)
++count;
}
int tmp = 0;
for(int i = 0; i < PIXEL_RANGE_MAX;++i)
{
tmp += histogram[i];
}
//printf("\n index %d, count %d, %d\n", index, count, tmp);

while(1)
{
if(count < FILTER_WINDOW_MEDIAN_SIZE)
{
count += histogram[index];
if(index < (PIXEL_RANGE_MAX-1))
++index;
else
break;
if(count > FILTER_WINDOW_MEDIAN_SIZE)
break;
}
if(count > FILTER_WINDOW_MEDIAN_SIZE)
{
count -= histogram[index-1];
if(index > 0)
--index;
else
break;
if(count < FILTER_WINDOW_MEDIAN_SIZE)
break;
}
if(count == FILTER_WINDOW_MEDIAN_SIZE)
break;
}

*pcurdst = index;
}

return;
}

int main(int argc, char* argv[])
{
int w = 0, h = 0;
std::string fileName = argv[1];
int iter = 1;
if(argc > 2)
iter = atoi(argv[2]);

cv::Mat src = cv::imread(fileName, CV_LOAD_IMAGE_GRAYSCALE);
cv::Mat dst ;
dst = src.clone();

printf("%d,%d,%d\n", src.rows, src.cols, src.channels());
imwrite("src.bmp", src);

for(int i = FILTER_WINDOW_Y/2; i < src.rows-FILTER_WINDOW_Y/2; i++)
{
unsigned char *pdst = dst.ptr<unsigned char>(i);
median_filter( src, i, pdst);
}

imwrite("dst.bmp", dst);

return 0;
}