jhj0517 commited on
Commit
c30ec73
1 Parent(s): bd8868d

fix output path

Browse files
Files changed (2) hide show
  1. musepose_inference.py +3 -1
  2. pose_align.py +3 -1
musepose_inference.py CHANGED
@@ -40,7 +40,9 @@ class MusePoseInference:
40
  self.pose_guider = None
41
  self.image_enc = None
42
  self.pipe = None
43
- self.output_dir = os.path.join("assets", "video")
 
 
44
  self.download_models()
45
 
46
  def infer_musepose(
 
40
  self.pose_guider = None
41
  self.image_enc = None
42
  self.pipe = None
43
+ self.output_dir = os.path.join("assets", "videos")
44
+ if not os.path.exists(self.output_dir):
45
+ os.makedirs(self.output_dir)
46
  self.download_models()
47
 
48
  def infer_musepose(
pose_align.py CHANGED
@@ -30,7 +30,9 @@ class PoseAlignmentInference:
30
  "pose_config": os.path.join("pose", "config", "dwpose-l_384x288.py"),
31
  "det_config": os.path.join("pose", "config", "yolox_l_8xb8-300e_coco.py"),
32
  }
33
- self.output_dir = os.path.join("assets", "video")
 
 
34
  self.download_models()
35
 
36
  def align_pose(
 
30
  "pose_config": os.path.join("pose", "config", "dwpose-l_384x288.py"),
31
  "det_config": os.path.join("pose", "config", "yolox_l_8xb8-300e_coco.py"),
32
  }
33
+ self.output_dir = os.path.join("assets", "videos")
34
+ if not os.path.exists(self.output_dir):
35
+ os.makedirs(self.output_dir)
36
  self.download_models()
37
 
38
  def align_pose(