Browse Source

[to #42322933] fix init issues for multi-style cartoon models

1. 修复多风格模型pipeline初始化问题
        Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10249429
master
myf272609 yingda.chen 3 years ago
parent
commit
047904ef73
1 changed files with 7 additions and 4 deletions
  1. +7
    -4
      modelscope/pipelines/cv/image_cartoon_pipeline.py

+ 7
- 4
modelscope/pipelines/cv/image_cartoon_pipeline.py View File

@@ -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'))


Loading…
Cancel
Save