feat(script): add entry point for image transform viz (#2007)

* feat(Scripts): add entry point for img transform viz

* chore(style): pre-commit style
This commit is contained in:
Steven Palma
2025-09-23 18:47:36 +02:00
committed by GitHub
parent c435d3cebc
commit c9787bd98a
4 changed files with 12 additions and 7 deletions

View File

@@ -20,10 +20,10 @@ Additionally, each individual transform can be visualized separately as well as
Example:
```bash
python -m lerobot.scripts.visualize_image_transforms \
--repo_id=lerobot/pusht \
--episodes='[0]' \
--image_transforms.enable=True
lerobot-imgtransform-viz \
--repo_id=lerobot/pusht \
--episodes='[0]' \
--image_transforms.enable=True
```
"""
@@ -126,5 +126,9 @@ def visualize_image_transforms(cfg: DatasetConfig, output_dir: Path = OUTPUT_DIR
save_each_transform(cfg.image_transforms, original_frame, output_dir, n_examples)
if __name__ == "__main__":
def main():
visualize_image_transforms()
if __name__ == "__main__":
main()