From 185b832ceef9fd8667d64587bcfd26e0f772f154 Mon Sep 17 00:00:00 2001 From: Marcin Kostrzewski Date: Wed, 1 Feb 2023 21:16:33 +0100 Subject: [PATCH] Fix debug flags --- main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 91ac921..e2be833 100644 --- a/main.py +++ b/main.py @@ -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):