如何从目录中找到与输入图像相似的图像？

import cv2
import numpy as np
from sklearn.metrics.pairwise import cosine_similarity
import os

def extract_features(image_path):
    img = cv2.imread(image_path)
    img = cv2.resize(img, (128, 128))
    img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    img = img.flatten()
    return img

def find_similar_images(query_image_path, image_directory, top_k=5):
    query_feature = extract_features(query_image_path)
    similarities = []
    
    for filename in os.listdir(image_directory):
        if filename == os.path.basename(query_image_path):
            continue
        image_path = os.path.join(image_directory, filename)
        feature = extract_features(image_path)
        similarity = cosine_similarity([query_feature], [feature])[0][0]
        similarities.append((filename, similarity))
    
    similarities.sort(key=lambda x: x[1], reverse=True)
    return similarities[:top_k]

# 示例使用
query_image_path = 'path_to_query_image.jpg'
image_directory = 'path_to_image_directory'
similar_images = find_similar_images(query_image_path, image_directory)
print(similar_images)