Merge branch 'main' of https://git.wmi.amu.edu.pl/s444498/wko_anime-face-similarity

.editorconfig

@@ -0,0 +1,3 @@
+[*.py]
+indent_style = space
+indent-size=2

main.py

@@ -0,0 +1,35 @@
+# Allows imports from the style transfer submodule
+import sys
+sys.path.append('DCT-Net')
+
+import cv2
+import os
+import numpy as np
+
+from source.cartoonize import Cartoonizer
+
+
+def load_source(filename: str) -> np.ndarray:
+  return cv2.imread(filename)[...,::-1]
+
+
+def find_and_crop_face(data: np.ndarray) -> np.ndarray:
+  # TODO
+  return data
+
+
+def compare_with_anime_characters(data: np.ndarray) -> int:
+  # TODO
+  return 1
+
+
+def transfer_to_anime(ima: np.ndarray):
+  algo = Cartoonizer(dataroot='damo/cv_unet_person-image-cartoon_compound-models')
+  return algo.cartoonize(img)
+
+
+if __name__ == '__main__':
+  source = load_source('input.png')
+  source_face = find_and_crop_face(source)
+  source_face_anime = transfer_to_anime(source)
+  print(compare_with_anime_characters(source_face_anime))