如何在Python中为图像创建KNN图(使用自定义距离度量)？

import numpy as np
import cv2
from sklearn.neighbors import NearestNeighbors

image = cv2.imread('image.jpg')
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
features = gray.reshape(-1, 1)

knn = NearestNeighbors(n_neighbors=5, metric='euclidean')
knn.fit(features)

def custom_distance(x, y):
    # 自定义距离度量逻辑
    return distance

knn.effective_metric_ = custom_distance
knn._fit_X = features

query_image = cv2.imread('query_image.jpg')
query_gray = cv2.cvtColor(query_image, cv2.COLOR_BGR2GRAY)
query_features = query_gray.reshape(-1, 1)

distances, indices = knn.kneighbors(query_features)