Added method outline

This commit is contained in:
Marcin Kostrzewski 2023-01-15 12:40:35 +01:00
parent 78b0f65fa0
commit f207e39c12
2 changed files with 38 additions and 0 deletions

3
.editorconfig Normal file
View File

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

35
main.py Normal file
View File

@ -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))