deepgeek/MuseTalk
2
0
Fork 0
mirror of https://github.com/TMElyralab/MuseTalk.git synced 2026-02-04 17:39:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
itechmusic
2024-04-02 19:37:48 +08:00
committed by czk32611
parent 4bb0398b53
commit 4c9c634fc4
6 changed files with 52 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/inference.py
View File

@@ -30,8 +30,8 @@ def main(args):
input_basename = os.path.basename(video_path).split('.')[0]
audio_basename = os.path.basename(audio_path).split('.')[0]
output_basename = f"{input_basename}_{audio_basename}"
crop_coord_save_path = os.path.join(args.result_dir, input_basename+".pkl") # only related to video input
result_img_save_path = os.path.join(args.result_dir, output_basename) # related to video & audio inputs
crop_coord_save_path = os.path.join(result_img_save_path, input_basename+".pkl") # only related to video input
os.makedirs(result_img_save_path,exist_ok =True)
if args.output_vid_name=="":
@@ -122,7 +122,7 @@ def main(args):
os.system(cmd_combine_audio)
os.system("rm temp.mp4")
os.system(f"rm -r {result_img_save_path}")
os.system(f"rm -rf {result_img_save_path}")
print(f"result is save to {output_vid_name}")
if __name__ == "__main__":