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

Plot results #6

Merged
s444409 merged 10 commits from pretty-results into main 2023-02-01 22:14:52 +01:00
Conversation 0 Commits 10 Files Changed 11 +2 -1
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit ca9163d134 - Show all commits

3
main.py
View File

@ -55,7 +55,8 @@ def get_top_results(all_metrics: list[dict], metric='correlation', count=1):
def transfer_to_anime(img: np.ndarray):
algo = Cartoonizer(dataroot='DCT-Net/damo/cv_unet_person-image-cartoon_compound-models')
return algo.cartoonize(img).astype(np.uint8)
model_out = algo.cartoonize(img).astype(np.uint8)
return cv2.cvtColor(model_out, cv2.COLOR_BGR2RGB)
def validate(test_set, anime_faces_set, metric='correlation'):