|
|
|
@@ -39,10 +39,13 @@ class ImageCartoonPipeline(Pipeline): |
|
|
|
super().__init__(model=model, **kwargs) |
|
|
|
with device_placement(self.framework, self.device_name): |
|
|
|
self.facer = FaceAna(self.model) |
|
|
|
self.sess_anime_head = self.load_sess( |
|
|
|
os.path.join(self.model, 'cartoon_h.pb'), 'model_anime_head') |
|
|
|
self.sess_anime_bg = self.load_sess( |
|
|
|
os.path.join(self.model, 'cartoon_bg.pb'), 'model_anime_bg') |
|
|
|
with tf.Graph().as_default(): |
|
|
|
self.sess_anime_head = self.load_sess( |
|
|
|
os.path.join(self.model, 'cartoon_h.pb'), |
|
|
|
'model_anime_head') |
|
|
|
self.sess_anime_bg = self.load_sess( |
|
|
|
os.path.join(self.model, 'cartoon_bg.pb'), |
|
|
|
'model_anime_bg') |
|
|
|
|
|
|
|
self.box_width = 288 |
|
|
|
global_mask = cv2.imread(os.path.join(self.model, 'alpha.jpg')) |
|
|
|
|