This commit is contained in:
unknown
2025-02-13 00:13:05 +02:00
parent 65058c8ef0
commit d050e1ac88
+2 -2
View File
@@ -29,7 +29,7 @@ def SwitchColors(image, detected_colors, target_colors, clustering_model, distan
distance_methods = {
"Euclidean": EuclideanDistance,
"Manhattan": ManhattanDistance,
"HSV_Similarity": CosineSimilarity
"Cosine Similarity": CosineSimilarity
}
distance_method = distance_methods.get(distance_method)
@@ -55,7 +55,7 @@ class PaletteTransferNode:
"target_colors": ("COLOR_LIST",),
"color_space": (["RGB", "HSV", "LAB"], {'default': 'RGB'}),
"cluster_method": (["Kmeans","Mini batch Kmeans"], {'default': 'Kmeans'}, ),
"distance_method": (["Euclidean", "Manhattan", "CosineSimilarity"], {'default': 'Euclidean'}, )
"distance_method": (["Euclidean", "Manhattan", "Cosine Similarity"], {'default': 'Euclidean'}, )
}
}
return data_in