Browse Source

[to #44902165] remove device placement for image cartoon to avoid full gpu memory usage

Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10260495
master
wenmeng.zwm 3 years ago
parent
commit
c8be0e8b78
1 changed files with 6 additions and 9 deletions
  1. +6
    -9
      modelscope/pipelines/cv/image_cartoon_pipeline.py

+ 6
- 9
modelscope/pipelines/cv/image_cartoon_pipeline.py View File

@@ -37,15 +37,12 @@ class ImageCartoonPipeline(Pipeline):
model: model id on modelscope hub.
"""
super().__init__(model=model, **kwargs)
with device_placement(self.framework, self.device_name):
self.facer = FaceAna(self.model)
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.facer = FaceAna(self.model)
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