尝试对ImageHash库使用差异散列时出现Numpy错误

Question

我尝试使用python库执行差分哈希运算，但一直收到ImageHash错误。

错误：

文件图像第252行，在dhash "/Users/testuser/Desktop/test_folder/test_env/lib/python3.8/site-packages/imagehash.py"，= image.convert("L").resize((hash_size + 1，hash_size)，Image.ANTIALIAS) AttributeError：'numpy.ndarray‘对象没有'convert’属性。

代码：

from PIL import Image
from cv2 import cv2
import imagehash
import numpy as np

def hash_and_compare(image1, image2):
    image1 = image1
    image2 = image2

    # read images
    image1 = cv2.imread(image1)
    image2 = cv2.imread(image2)

    # convert to grayscale
    image1 = cv2.cvtColor(image1, cv2.COLOR_BGR2GRAY)
    image2 = cv2.cvtColor(image2, cv2.COLOR_BGR2GRAY)

    # resize images
    image1 = cv2.resize(image1, (9, 8))
    image2 = cv2.resize(image2, (9, 8))

    # hash images
    image1_hash = imagehash.dhash(image1)
    image2_hash = imagehash.dhash(image2)

    # compute hamming distance
    distance = image1_hash - image2_hash

    if image1_hash <= 10:
        print(distance)
        print('match')
    else:
        print(distance)
        print('no match')

hash_and_compare('/Users/testuser/Desktop/test_folder/game_name056496.png', '/Users/testuser/Desktop/test_folder/game_name499761.png')

Answer 1

正如imagehash库的文档@image must be a PIL instance.中所提到的。所以你不能设置numpy数组作为dshash function.if的输入，你想用opencv做一些预处理，你应该在设置成dhash之前把它转换成PIL数组，如下所示：

import numpy as np
from PIL import Image
...
some preprocess
...
# You may need to convert the color.
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
im_pil = Image.fromarray(img)
image1_hash = imagehash.dhash(im_pil)