From 8237ed9aa461943ba76f64b8e378f792d24b94c2 Mon Sep 17 00:00:00 2001 From: Marina Barannikov Date: Wed, 5 Jun 2024 16:01:37 +0000 Subject: [PATCH] Updated visualize script --- lerobot/scripts/visualize_image_transforms.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/lerobot/scripts/visualize_image_transforms.py b/lerobot/scripts/visualize_image_transforms.py index a833117b..49448838 100644 --- a/lerobot/scripts/visualize_image_transforms.py +++ b/lerobot/scripts/visualize_image_transforms.py @@ -35,9 +35,6 @@ def main(cfg, output_dir=Path("outputs/image_transforms")): # Apply each single transformation for transform_name in transforms: - cfg.image_transform.enable = True - cfg.image_transform.max_num_transforms = 1 - for t in transforms: if t == transform_name: cfg.image_transform[t].weight = 1