Added method outline
This commit is contained in:
parent
78b0f65fa0
commit
f207e39c12
3
.editorconfig
Normal file
3
.editorconfig
Normal file
@ -0,0 +1,3 @@
|
||||
[*.py]
|
||||
indent_style = space
|
||||
indent-size=2
|
35
main.py
Normal file
35
main.py
Normal 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))
|
Loading…
Reference in New Issue
Block a user