From 4909c8f48abbe04006fc9c233023c70185f31af0 Mon Sep 17 00:00:00 2001 From: jaydrennan Date: Wed, 22 Nov 2023 17:47:43 -0800 Subject: [PATCH] feature: adds details to video file name generation adds seed, model, and original image name into video file name --- imaginairy/video_sample.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/imaginairy/video_sample.py b/imaginairy/video_sample.py index c8ceb1c2..3e2ebf6b 100644 --- a/imaginairy/video_sample.py +++ b/imaginairy/video_sample.py @@ -186,7 +186,9 @@ def denoiser(_input, sigma, c): os.makedirs(output_folder, exist_ok=True) base_count = len(glob(os.path.join(output_folder, "*.mp4"))) - video_path = os.path.join(output_folder, f"{base_count:06d}.mp4") + input_file_name = os.path.basename(input_img_path) + video_path = os.path.join(output_folder, f"{base_count:06d}_{model_name}_{str(seed)}_{input_file_name}.mp4") + writer = cv2.VideoWriter( video_path, cv2.VideoWriter_fourcc(*"MP4V"),