fix
This commit is contained in:
+2
-2
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user