From 4a8c5e238e2388abe632287837d713488f084548 Mon Sep 17 00:00:00 2001 From: Cadene Date: Tue, 26 Mar 2024 10:12:16 +0000 Subject: [PATCH] issue with cat_and_write_video --- lerobot/scripts/visualize_dataset.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/lerobot/scripts/visualize_dataset.py b/lerobot/scripts/visualize_dataset.py index 7613b68d..3dd7cdfa 100644 --- a/lerobot/scripts/visualize_dataset.py +++ b/lerobot/scripts/visualize_dataset.py @@ -27,8 +27,7 @@ def visualize_dataset_cli(cfg: dict): def cat_and_write_video(video_path, frames, fps): # Expects images in [0, 255]. frames = torch.cat(frames) - assert frames.max() <= 1 and frames.min() >= 0 - frames = (255 * frames).to(dtype=torch.uint8) + assert frames.dtype == torch.uint8 frames = einops.rearrange(frames, "b c h w -> b h w c").numpy() imageio.mimsave(video_path, frames, fps=fps)