Fixes #8
2
main.py
2
main.py
@ -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)
|
||||||
|
Loading…
Reference in New Issue
Block a user