搜索和很久,还是没有发现求mat 内元素的中值函数,于是自己写了一个
float Median_Mat_32f(Mat img)
{
float *buf;
buf = new float[img.rows*img.cols];
for (int i =0; i < img.rows; i++)
{
for (int j = 0; j < img.cols; j++)
{
buf[i*img.cols+j] = img.ptr<float>(i)[j];
}
}
qsort(buf, 3, sizeof(buf[0]), comp);
return buf[img.rows*img.cols/2];
}
数据类型不确定,于是又想起了写一个模板
上代码:
//比较两数大小
template <typename _Tp>
int mem_cmp(const void *a, const void *b)
{
//当_Tp为浮点型,可能由于精度,会影响排序
return (*((_Tp *)a) - *((_Tp *)b));
}
//求Mat元素中值
template <typename _Tp>
_Tp medianElem(Mat img)
{
_Tp *buf;
size_t total = img.total();
buf = new _Tp[total];
for (int i = 0; i < img.rows; i++)
{
for (int j = 0; j < img.cols; j++)
{
buf[i*img.cols+j] = img.ptr<_Tp>(i)[j];
}
}
qsort(buf, total, sizeof(_Tp), mem_cmp<_Tp>);
return buf[total/2];
}
//求Mat元素总和(单通道)
template <typename _Tp>
double sumElem(Mat img)
{
double sum = 0;
for (int i = 0; i < img.rows; i++)
{
for (int j = 0; j < img.cols; j++)
{
sum += img.ptr<_Tp>(i)[j];
}
}
return sum;
}
//求Mat元素均值(单通道)
template <typename _Tp>
_Tp sumElem(Mat img)
{
return _Tp(sumElem<_Tp>(img)/img.total());
}
这样就好用多了。
欢迎拍砖~