opencv Mat 求元素 中值 均值 总和

搜索和很久，还是没有发现求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());
}

这样就好用多了。

欢迎拍砖~