From 047904ef73d42eccb33328089642ae6ffe20318d Mon Sep 17 00:00:00 2001 From: myf272609 Date: Mon, 26 Sep 2022 11:55:06 +0800 Subject: [PATCH] [to #42322933] fix init issues for multi-style cartoon models MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 1. 修复多风格模型pipeline初始化问题 Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10249429 --- modelscope/pipelines/cv/image_cartoon_pipeline.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/modelscope/pipelines/cv/image_cartoon_pipeline.py b/modelscope/pipelines/cv/image_cartoon_pipeline.py index f34be618..72fda989 100644 --- a/modelscope/pipelines/cv/image_cartoon_pipeline.py +++ b/modelscope/pipelines/cv/image_cartoon_pipeline.py @@ -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'))