From 774e9bcc9224e57fdfe2cc3481e7e0ace3e41e89 Mon Sep 17 00:00:00 2001 From: yaohwang Date: Wed, 1 Feb 2023 15:22:17 +0800 Subject: [PATCH] fix issue #28 --- source/cartoonize.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/source/cartoonize.py b/source/cartoonize.py index 9c9a42b..f38bddd 100644 --- a/source/cartoonize.py +++ b/source/cartoonize.py @@ -16,9 +16,9 @@ class Cartoonizer(): self.facer = FaceAna(dataroot) self.sess_head = self.load_sess( - os.path.join(dataroot, 'cartoon_anime_h.pb'), 'model_head') + os.path.join(dataroot, 'cartoon_h.pb'), 'model_head') self.sess_bg = self.load_sess( - os.path.join(dataroot, 'cartoon_anime_bg.pb'), 'model_bg') + os.path.join(dataroot, 'cartoon_bg.pb'), 'model_bg') self.box_width = 288 global_mask = cv2.imread(os.path.join(dataroot, 'alpha.jpg'))