From 37925c25fb6be564f19308451a84beb2ac2fbac1 Mon Sep 17 00:00:00 2001 From: Marcin Kostrzewski Date: Wed, 1 Feb 2023 21:27:39 +0100 Subject: [PATCH] Use correlation --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index e2be833..4301743 100644 --- a/main.py +++ b/main.py @@ -104,7 +104,7 @@ def main(): source = load_source('test_set/Ayanokouji, Kiyotaka.jpg') results = similarity_to_anime(source, anime_faces_set) - method = 'structural-similarity' + method = 'correlation' top_results = get_top_results(results, count=4, metric=method) print(top_results) plot_results(source, transfer_to_anime(source), top_results, anime_faces_set, method)