s444498/wko_anime-face-similarity
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use correlation

Browse Source
This commit is contained in:
Marcin Kostrzewski 2023-02-01 21:27:39 +01:00
parent 185b832cee
commit 37925c25fb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@ -104,7 +104,7 @@ def main():
source = load_source('test_set/Ayanokouji, Kiyotaka.jpg') source = load_source('test_set/Ayanokouji, Kiyotaka.jpg')
results = similarity_to_anime(source, anime_faces_set) results = similarity_to_anime(source, anime_faces_set)
method = 'structural-similarity' method = 'correlation'
top_results = get_top_results(results, count=4, metric=method) top_results = get_top_results(results, count=4, metric=method)
print(top_results) print(top_results)
plot_results(source, transfer_to_anime(source), top_results, anime_faces_set, method) plot_results(source, transfer_to_anime(source), top_results, anime_faces_set, method)