本博客中使用的代码见本文末尾

       度量两张图片的相似度有许多算法,本文讲介绍工程领域中最常用的图片相似度算法之一——Hash算法。Hash算法准确的说有三种,分别为平均哈希算法(aHash)、感知哈希算法你(pHash)和差异哈哈希算法(dHash)。

       三种Hash算法都是通过获取图片的hash值,再比较两张图片hash值的汉明距离(韩明距离的概念可见本公众号《》一文)来度量两张图片是否相似。两张图片越相似,那么两张图片的hash数的汉明距离越小。下面本文将分别介绍这三种Hash算法。

1 平均哈希算法(aHash)

1.1 算法步骤

       平均哈希算法是三种Hash算法中最简单的一种,它通过下面几个步骤来获得图片的Hash值,这几个步骤分别是(1) 缩放图片;(2)转灰度图; (3) 算像素均值;(4)根据相似均值计算指纹。具体算法如下所示:

       得到图片的ahash值后,比较两张图片ahash值的汉明距离,通常认为汉明距离小于10的一组图片为相似图片。

1.2 具体实例

       本文图片为Lena图来说明.

       其中转为8x8尺寸的Lena对应的数据矩阵为:

       很容得到如上矩阵所有元素的均值a= 121.328125, 将上述矩阵中大于或等于a的元素置为1, 小于a的元素置为0,可得:

       所以可得Lena图的aHash为

       1011111010011110100111011010100110101011101000110000111000101100

       将二进制形式ahash转十六进制hash为

       be9e9da9aba30e2c

       为了测试aHash算法的效果,我们用一张带噪声Lena(noise)图和与Lena不一样的Barbara做图片相似度对比实验,其中Lena(noise)和Barbara如下:

       通过aHash算法容易得三个图片的hash值,然后根据hanming距离计算Lena(origin).png和Lena(noise).png Barbar.png之间汉明距离,具体如下:

       由上图可见aHash能区别相似图片和差异大的图片。

2 感知哈希算法(pHash)

2.1 算法步骤

       感知哈希算法是三种Hash算法中较为复杂的一种,它是基于DCT(离散余弦变换)来得到图片的hash值,其算法几个步骤分别是(1) 缩放图片;(2)转灰度图; (3) 计算DCT;(4)缩小DCT; (5)算平均值;(6) 计算指纹。具体算法如下所示:

       得到图片的phash值后,比较两张图片phash值的汉明距离,通常认为汉明距离小于10的一组图片为相似图片。

2.2 具体实例

       仍用Lena图来说明.

       通过计算可得灰度32x32Lenna图对应的DCT矩阵左上角8x8区域子矩阵为:

       很容得到如上矩阵所有元素的均值a= 77.35, 将上述矩阵中大于或等于a的元素置为1, 小于a的元素置为0,可得:

       所以可得Lena图的pHash为

       1001100111000100010101000010010101100000001000111000001010000000

       将二进制形式phash转十六进制hash为

       99c4542560238280

       为了测试pHash算法的效果,同样用一张带噪声Lena(noise)图和与Lena不一样的Barbara做图片相似度对比实验。通过pHash算法容易得三个图片的hash值,然后根据hanming距离计算Lena(origin).png和Lena(noise).png Barbar.png之间汉明距离,具体如下:

       由上图可见pHash能区别相似图片和差异大的图片。

3 差异哈希算法(dHash)

3.1 算法步骤

       相比pHash,dHash的速度要快的多,相比aHash,dHash在效率几乎相同的情况下的效果要更好,它是基于渐变实现的。其算法几个步骤分别是(1) 缩放图片;(2)转灰度图; (3) 计算DCT;
(4)缩小DCT; (5)算平均值;(6) 计算指纹。具体算法如下所示:

       得到图片的phash值后,比较两张图片phash值的汉明距离,通常认为汉明距离小于10的一组图片为相似图片。

3.2 具体实例

       仍用Lena图来说明.

       通过计算可得灰度9x8Lenna图数据矩阵为:

       从第二行开始进行减去前一行操作,可得如下查分矩阵

       将上述矩阵中大于或等于0元素置为1, 小于a的元素置为0,可得:

       所以可得Lena图的dHash为

       0101100000110111111010000101001001101011101011110001010001010000

       将二进制形式dhash转十六进制hash为

       99c4542560238280

       为了测试dHash算法的效果,同样用一张带噪声Lena(noise)图和与Lena不一样的Barbara做图片相似度对比实验。通过pHash算法容易得三个图片的hash值,然后根据hanming距离计算Lena(origin).png和Lena(noise).png Barbar.png之间汉明距离,具体如下:

       由上图可见dHash能区别相似图片和差异大的图片。

总结

       关于图像相似度算法除了Hash算法,在传统算法领域中还有基于SIFT的匹配算法,基于Gist特征的匹配算法;在深度学习领域中有基于ResNet全连接的匹配算法。感兴趣的读者可以通过google来了解这些算法。

参考资料

432-Looks-Like-It
529-Kind-of-Like-That

本文代码

Github代码

07-29 05:43