36 lines
793 B
Python
36 lines
793 B
Python
|
# 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))
|