yerang commited on
Commit
ce87533
·
verified ·
1 Parent(s): 01f8cd1

Update stf_utils.py

Browse files
Files changed (1) hide show
  1. stf_utils.py +7 -0
stf_utils.py CHANGED
@@ -80,9 +80,16 @@ class STFPipeline:
80
  ):
81
  #os.makedirs(root_path, exist_ok=True)
82
  import shutil; shutil.copytree('/home/user/app/stf/works', '/tmp/works', dirs_exist_ok=True)
 
 
83
  dir_zip='/tmp/works/preprocess/nasilhong_f_v1_front/crop_video_front_one_piece_dress_nodded_cut.zip'
84
  dir_target='/tmp/works/preprocess/nasilhong_f_v1_front/'
 
 
 
 
85
  import zipfile; zipfile.ZipFile(dir_zip, 'r').extractall(dir_target)
 
86
 
87
  self.config_path = os.path.join(stf_path, config_path)
88
  self.checkpoint_path = os.path.join(stf_path, checkpoint_path)
 
80
  ):
81
  #os.makedirs(root_path, exist_ok=True)
82
  import shutil; shutil.copytree('/home/user/app/stf/works', '/tmp/works', dirs_exist_ok=True)
83
+
84
+ import zipfile
85
  dir_zip='/tmp/works/preprocess/nasilhong_f_v1_front/crop_video_front_one_piece_dress_nodded_cut.zip'
86
  dir_target='/tmp/works/preprocess/nasilhong_f_v1_front/'
87
+ zipfile.ZipFile(dir_zip, 'r').extractall(dir_target)
88
+
89
+ dir_zip='/tmp/works/preprocess/nasilhong_f_v1_front/front_one_piece_dress_nodded_cut.zip'
90
+ dir_target='/tmp/works/preprocess/nasilhong_f_v1_front/'
91
  import zipfile; zipfile.ZipFile(dir_zip, 'r').extractall(dir_target)
92
+
93
 
94
  self.config_path = os.path.join(stf_path, config_path)
95
  self.checkpoint_path = os.path.join(stf_path, checkpoint_path)