根据直方图计算图像间的互信息

本文根据博客《互信息图像配准》https://blog.csdn.net/wuxingyu_cs/article/details/50397280,使用opencv对算法进行实现。

环境:VS2013+OpenCV3.0

网上流传较多的计算图像熵及互信息代码均是通过计算每一个像素点灰度值概率实现的,个人认为这种算法与图像熵或互信息原理不符(也可能是因为新手,理解有误),因此编写了一个使用灰度直方图计算图像熵与互信息的方法,如有错误,敬请指正。

废话少说,直接上代码。

计算图像熵,输入为灰度图像及灰度等级

double Entorpy(Mat& inputImg,int size)
{
    MatND dstHist;       
    int dims = 1;
    float hranges[] = { 0, 255 };
    const float *ranges[] = { hranges };   
    int channels = 0;
    calcHist(&inputImg, 1, 0, Mat(), dstHist, dims, &size, ranges);    
    double Ent = 0;
    for (int i = 0; i < 16; i++)
    {
        double value = (double)dstHist.at<float>(i);
        double pr = value / (inputImg.rows*inputImg.cols);
        if (pr > 0)
        {
            Ent -= pr * (log(pr) / log(2.0));//熵
        }
    }
    return Ent;
}

计算图像互信息,输入两幅待匹配灰度图像。

double CoEntorpy(Mat& inputImg1, Mat& inputImg2)
{
    double CoEnt = 0;

    MatND inpuImg1_dstHist, inpuImg2_dstHist;      
    int dims = 1;
    float hranges[] = { 0, 255 };
    const float *ranges[] = { hranges };   
    int size = 64;      //灰度等级
    int channels = 0;
    calcHist(&inputImg1, 1, 0, Mat(), inpuImg1_dstHist, dims, &size, ranges);    
    calcHist(&inputImg2, 1, 0, Mat(), inpuImg2_dstHist, dims, &size, ranges);

    for (int i = 0; i < size; i++)
    {
        double inpuImg1_value = (double)inpuImg1_dstHist.at<float>(i);
        double inpuImg1_pr = inpuImg1_value / (inputImg1.rows*inputImg1.cols);
        double inpuImg2_value = (double)inpuImg2_dstHist.at<float>(i);
        double inpuImg2_pr = inpuImg2_value / (inputImg2.rows*inputImg2.cols);
        double pr = (inpuImg1_value + inpuImg2_value) / (inputImg1.rows*inputImg1.cols + inputImg2.rows*inputImg2.cols);

        if (pr > 0 && inpuImg1_pr > 0 && inpuImg2_pr > 0)
        {
            CoEnt += pr * (log(pr / (inpuImg1_pr*inpuImg2_pr)) / log(2.0));//熵
        }
    }

    return  CoEnt;
}

 

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值