1. 程式人生 > >opencv實現matlab的median()函式

opencv實現matlab的median()函式

總結一下之前在做專案時需要用到把matlab的一些程式碼轉換成c/C++程式碼,下面是實現matlab中的median()函式的程式碼,供大家借鑑。
matlab中median()函式的實現。
int cmp(const void a, const void b)//排序函式,
{
return (float )a>(float)b ? 1 : -1;
}
///////////
int median(vector<IplImage
> LR_I, IplImage
result)
{
int LR_value = LR_I.size();
float last_gray = 0;
float p[20] = { 0 };
for (int i = 0;i< LR_I[0]->width; i++)
{
for (int j = 0;j< LR_I[0]->height; j++)
{
for (int k = 0; k < LR_value; k++)
{
float gray = cvGetReal2D(LR_I[k], i, j);
p[k] = gray;
}
qsort(p, LR_value, sizeof(p[0]), cmp);
if (LR_value%2==0)
{
int m = LR_value / 2;
int n = (LR_value / 2)-1;
last_gray = (p[m] + p[n]) / 2.0;
}
else
{
int m = LR_value / 2;
last_gray = p[m];
}
cvSetReal2D(result, i, j, last_gray);
}
}
return 0;
}

這裡需要注意的是排序函式qsort(),在對float或是double型進行排序時,要用三目運算子,
寫返回值,return *(int )a >(int *)b?1:-1; 因為cmp排序函式返回值為int型,如果兩個數很相近,則會認為返回值為0,無法正確排序。
//整型陣列排序
int num[100];
int cmp ( const void *a , const void *b )
{
return *(int *)a - *(int *)b;
}
qsort(num,100,sizeof(num[0]),cmp);
///double型陣列排序
Double num[100];
int cmp ( const void a , const void b )
{
return (double

)a >(double)b? 1:-1;
}
qsort(num,100,sizeof(num[0]),cmp);