File size: 690 Bytes
528f5ef |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
import numpy as np
from sklearn.neighbors import NearestNeighbors
# Convert the feature vectors to a NumPy array
features_array = np.array(list(features_dict.values()))
# Create a NearestNeighbors object and fit it to the features array
knn = NearestNeighbors(n_neighbors=11, metric='cosine')
knn.fit(features_array)
# Define a function to retrieve the most similar images to a query image
def retrieve_similar_images(query_image_name, knn_model, features_dict):
# Get the features for the query image
query_features = features_dict[query_image_name]
# Reshape the features to match the expected input format for the knn_model
query_features = query_features.reshape(1, - |