From 995f3ffd8dd082f2261f9c1cbd73c9644532f5bf Mon Sep 17 00:00:00 2001 From: Somdev Sangwan Date: Mon, 29 May 2023 17:42:52 +0530 Subject: [PATCH] keep frames in select folder --- run.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/run.py b/run.py index f928768..184a80a 100644 --- a/run.py +++ b/run.py @@ -66,13 +66,12 @@ def start(): return global pool pool = mp.Pool(psutil.cpu_count()-1) - current_dir = os.getcwd() target_path = args['target_path'] if is_img(target_path): process_img(args['source_img'], target_path) return video_name = target_path.split("/")[-1].split(".")[0] - output_dir = current_dir + "/" + video_name + output_dir = target_path.replace(target_path.split("/")[-1], "") + "/" + video_name Path(output_dir).mkdir(exist_ok=True) fps = detect_fps(target_path) if not args['keep_fps'] and fps > 30: @@ -88,7 +87,7 @@ def start(): )) start_processing() create_video(video_name, fps, output_dir) - add_audio(current_dir, output_dir, target_path, args['keep_frames'], args['output_file']) + add_audio(output_dir, target_path, args['keep_frames'], args['output_file']) save_path = args['output_file'] if args['output_file'] else output_dir + "/" + video_name + ".mp4" print("\n\nVideo saved as:", save_path, "\n\n")