相似度计算公式

小女孩真可爱

于 2022-07-16 23:28:39 发布

阅读量169

点赞数

文章标签： python numpy 机器学习

本文链接：https://blog.csdn.net/m0_48095841/article/details/125777406

版权

import os
import cv2
from PIL import Image
from numpy import average, linalg, dot


def get_thumbnail(image, size=(512, 512), greyscale=False):
    image = image.resize(size, Image.ANTIALIAS)
    if greyscale:
        image = image.convert('L')
    return image


def image_similarity_vectors_via_numpy(image1, image2):
    image1 = get_thumbnail(image1)
    image2 = get_thumbnail(image2)
    images = [image1, image2]
    vectors = []
    norms = []
    for image in images:
        vector = []
        for pixel_tuple in image.getdata():
            vector.append(average(pixel_tuple))
        vectors.append(vector)
        norms.append(linalg.norm(vector, 2))
    a, b = vectors
    a_norm, b_norm = norms
    res = dot(a / a_norm, b / b_norm)
    return res