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

Plot results #6

Merged
s444409 merged 10 commits from pretty-results into main 2023-02-01 22:14:52 +01:00
Conversation 0 Commits 10 Files Changed 11 +2 -2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 185b832cee - Show all commits

4
main.py
View File

@ -48,7 +48,7 @@ def transfer_to_anime(img: np.ndarray):
return cv2.cvtColor(model_out, cv2.COLOR_BGR2RGB)
def similarity_to_anime(source_image, anime_faces_set, debug=True):
def similarity_to_anime(source_image, anime_faces_set, debug=False):
try:
source_face_bbox = find_face_bbox(source_image)
except ValueError:
@ -69,7 +69,7 @@ def similarity_to_anime(source_image, anime_faces_set, debug=True):
plt.imshow(source_face_anime)
plt.show()
return compare_with_anime_characters(source_face_anime, anime_faces_set)
return compare_with_anime_characters(source_face_anime, anime_faces_set, verbose=debug)
def validate(test_set, anime_faces_set):