Updated visualize script

2024-06-05 16:01:37 +00:00
parent 82e32f1fcd
commit 8237ed9aa4
1 changed files with 0 additions and 3 deletions
--- 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