2023-02-01 22:14:44 +01:00
1 changed files with 1 additions and 1 deletions
--- 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)