jhj0517 commited on
Commit
606190d
1 Parent(s): 54af6fd
Files changed (1) hide show
  1. pose_align.py +3 -0
pose_align.py CHANGED
@@ -42,6 +42,7 @@ class PoseAlignmentInference:
42
  align_frame: int,
43
  max_frame: int,
44
  ):
 
45
  dt_file_name = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
46
  outfn=os.path.abspath(os.path.join(self.output_dir, f'{dt_file_name}_demo.mp4'))
47
  outfn_align_pose_video=os.path.abspath(os.path.join(self.output_dir, f'{dt_file_name}.mp4'))
@@ -312,12 +313,14 @@ class PoseAlignmentInference:
312
  def download_models(self):
313
  repo_id = 'jhj0517/MusePose'
314
  for name, file_path in self.model_paths.items():
 
315
  local_dir, filename = os.path.dirname(file_path), os.path.basename(file_path)
316
  if not os.path.exists(local_dir):
317
  os.makedirs(local_dir)
318
 
319
  remote_filepath = os.path.join("dwpose", filename)
320
  if not os.path.exists(file_path):
 
321
  hf_hub_download(repo_id=repo_id, filename=remote_filepath,
322
  local_dir=local_dir,
323
  local_dir_use_symlinks=False)
 
42
  align_frame: int,
43
  max_frame: int,
44
  ):
45
+ self.download_models()
46
  dt_file_name = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
47
  outfn=os.path.abspath(os.path.join(self.output_dir, f'{dt_file_name}_demo.mp4'))
48
  outfn_align_pose_video=os.path.abspath(os.path.join(self.output_dir, f'{dt_file_name}.mp4'))
 
313
  def download_models(self):
314
  repo_id = 'jhj0517/MusePose'
315
  for name, file_path in self.model_paths.items():
316
+
317
  local_dir, filename = os.path.dirname(file_path), os.path.basename(file_path)
318
  if not os.path.exists(local_dir):
319
  os.makedirs(local_dir)
320
 
321
  remote_filepath = os.path.join("dwpose", filename)
322
  if not os.path.exists(file_path):
323
+ print(file_path)
324
  hf_hub_download(repo_id=repo_id, filename=remote_filepath,
325
  local_dir=local_dir,
326
  local_dir_use_symlinks=False)