From 44bae5b03029c0acaa2119d9022e5059400e3a77 Mon Sep 17 00:00:00 2001 From: Hugo Zanini Date: Sat, 30 Dec 2023 10:39:30 -0300 Subject: [PATCH] Update util.py for printing the folder where the video is being saved --- magicanimate/utils/util.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/magicanimate/utils/util.py b/magicanimate/utils/util.py index 76f2245c..2124f3a3 100644 --- a/magicanimate/utils/util.py +++ b/magicanimate/utils/util.py @@ -31,6 +31,7 @@ def save_videos_grid(videos: torch.Tensor, path: str, rescale=False, n_rows=6, f os.makedirs(os.path.dirname(path), exist_ok=True) imageio.mimsave(path, outputs, fps=fps) + print("Video saved at ", path) def save_images_grid(images: torch.Tensor, path: str): assert images.shape[2] == 1 # no time dimension @@ -135,4 +136,4 @@ def slerp( #logger.info(f'warning: v0 and v1 close to parallel, using linear interpolation instead.') return (1.0 - t) * v0 + t * v1 omega = dot.acos() - return (((1.0 - t) * omega).sin() * v0 + (t * omega).sin() * v1) / omega.sin() \ No newline at end of file + return (((1.0 - t) * omega).sin() * v0 + (t * omega).sin() * v1) / omega.sin()