Я реализовал 2D-медианный фильтр на C. Для изображения размером 1440X1440 значения с плавающей запятой. Для начала я попробовал это с простым размером ядра 3X3. Вот код.
#define kernelSize 3
void sort(float *array2sort, int n)
{
float temp;
for(int i=0; i < n-1; i++)
for(int j=0; j < n-1-i; j++)
if(array2sort[j] > array2sort[j+1])
{
temp = array2sort[j];
array2sort[j] = array2sort[j+1];
array2sort[j+1] = temp;
}
}
void medianFilter(float *input, float *output)
{
int halfKernelSize = kernelSize/2;
float neighbourhood[kernelSize*kernelSize];
for(int i=0+halfKernelSize; i<(1440-halfKernelSize); i++)
for(int j=0+halfKernelSize; j<(1440-halfKernelSize); j++)
{
for(int ii=-halfKernelSize; ii<halfKernelSize+1; ii++)
for(int jj=-halfKernelSize; jj<halfKernelSize+1; jj++)
neighbourhood[(ii+halfKernelSize)*kernelSize+(jj+halfKernelSize)] = input[(i+ii)*1440+(j+jj)];
sort(neighbourhood, kernelSize*kernelSize);
output[(i)*1440+(j)] = neighbourhood[(kernelSize*kernelSize)/2+1];
}
}
Теперь, чтобы проверить, в порядке ли код, я взял изображение, добавил к нему шум соли и перца с помощью MATLAB. Затем попробовал приведенный выше код на нем. Я вижу, что шум уменьшается ПОЧТИ полностью, остается несколько точек. Если я увеличу размер ядра до 5X5, шум полностью отфильтруется. Но меня беспокоит тот факт, что код медианного фильтра MATLAB способен полностью удалить шум даже с ядром размером 3X3. Это оставляет меня в сомнении. Пожалуйста, взгляните на код и дайте мне знать, есть ли какая-то ошибка в реализации фильтра или код MATLAB выполняет какие-то дополнительные шаги.