wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Updated visualize script

This commit is contained in:
Marina Barannikov 2024-06-05 16:01:37 +00:00
parent 82e32f1fcd
commit 8237ed9aa4
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lerobot/scripts/visualize_image_transforms.py
View File

@ -35,9 +35,6 @@ def main(cfg, output_dir=Path("outputs/image_transforms")):
# Apply each single transformation # Apply each single transformation
for transform_name in transforms: for transform_name in transforms:
cfg.image_transform.enable = True
cfg.image_transform.max_num_transforms = 1
for t in transforms: for t in transforms:
if t == transform_name: if t == transform_name:
cfg.image_transform[t].weight = 1 cfg.image_transform[t].weight = 1