python计算两个图像的互信息

最新推荐文章于 2024-08-08 05:07:55 发布

sihaiyinan

最新推荐文章于 2024-08-08 05:07:55 发布

阅读量1w

点赞数 13

分类专栏： Python 文章标签： python 互信息 numpy

本文链接：https://blog.csdn.net/sihaiyinan/article/details/112196356

版权

本文介绍了如何使用Python的numpy和sklearn库计算两个图像之间的互信息，以此评估它们的相似程度。通过提供的代码示例，展示了两种计算方法并验证了结果的一致性。

摘要由CSDN通过智能技术生成

我们常用互信息来衡量两个图像的相似程度。互信息的计算方式如下：

使用python中的numpy包或者sklearn可以很方便的计算互信息，计算代码如下：


import cv2
import numpy as np
import sklearn.metrics as skm


def hxx_forward(x, y):
    return skm.mutual_info_score(x, y)

def hxx(x, y):
    size = x.shape[-1]
    px = np.histogram(x, 256, (0, 255))[0] / size
    py = np.histogram(y, 256, (0, 255))[0] / size
    hx = - np.sum(px * np.log(px + 1e-8))
    hy = - np.sum(py * np.log(py + 1e-8))

    hxy = np.histogram2d(x, y, 256, [[0, 255], [0, 255]])[0]
    hxy /= (1.0 * size)
    hxy = - np.sum(hxy * np.log(hxy + 1e-8))

    r = hx + hy - hxy
    return r


img1 = cv2.imread("F:/BYSY/ACCESS/ImageMatch/1.png", cv2.IMREAD_GRAYSCALE)
img2 = cv2.imread("F:/BYSY/ACCESS/ImageMatch/1.png", cv2.IMREAD_GRAYSCALE)

x = np.reshape(img1, -1)
y = np.reshape(img2, -1)

print(hxx_forward(x, y))
print(hxx(x, y))

经验证，两种方法计算结果相同