s444498/wko_anime-face-similarity
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add missing stuff

Browse Source
This commit is contained in:
Marcin Kostrzewski 2023-01-29 21:23:11 +01:00
parent 5c087fdcf2
commit 8e4c805163
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
main.py
View File

@ -1,16 +1,15 @@
# Allows imports from the style transfer submodule
import sys
sys.path.append('DCT-Net')
import cv2
import os
import numpy as np
# Allows imports from the style transfer submodule
sys.path.append('DCT-Net')
from source.cartoonize import Cartoonizer
def load_source(filename: str) -> np.ndarray:
return cv2.imread(filename)[...,::-1]
return cv2.imread(filename)[..., ::-1]
def find_and_crop_face(data: np.ndarray) -> np.ndarray:
@ -29,7 +28,7 @@ def compare_with_anime_characters(data: np.ndarray) -> int:
def transfer_to_anime(img: np.ndarray):
algo = Cartoonizer(dataroot='damo/cv_unet_person-image-cartoon_compound-models')
algo = Cartoonizer(dataroot='DCT-Net/damo/cv_unet_person-image-cartoon_compound-models')
return algo.cartoonize(img)

3
requirements.txt
View File

@ -5,4 +5,5 @@ modelscope==1.1.3
requests==2.28.2
beautifulsoup4==4.11.1
lxml==4.9.2
opencv-python==4.7.0.68
opencv-python==4.7.0.68
torch==1.13.1