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()