Add create reward visualization by pkooij · Pull Request #3155 · huggingface/lerobot

pkooij · 2026-03-13T16:34:23Z

Title

Add reward visualization also used in blog post

Checklist (required before merge)

Linting/formatting run (pre-commit run -a)
All tests pass locally (pytest)
Documentation updated
CI is green

…e/lerobot into feat/analysis_dataset

s1lent4gnt

Looks nice, but I would love to consider these comments:

Consider placing create_progress_videos.py at the same level as other scripts to stay consistent.
Add return types to all functions for consistency.
Add Args and Returns in functions docstring would help readability.
Use descriptive variable names, the script is simple but it would be nice if you stay consistent with lerobot codebase.
OpenCV can seek directly into the original video with cap.set(cv2.CAP_PROP_POS_MSEC, from_ts * 1000), eliminating one full decode/encode cycle in clip extraction from chunk and the intermediate file.
Consider making MP4 the default output with GIF as an optional flag.
The clip is opened three times: in composite_video, All three should be consolidated into a single open.

- Add shebang and Apache 2.0 license header - Replace hardcoded absolute OUTPUT_DIR with relative default (./progress_videos) - Add argparse CLI (--repo-id, --episode, --camera-key, --output-dir, --gif) - Wrap entrypoint in def main() - Replace all print() with logging - Use logging.error/warning instead of traceback.print_exc - Release VideoCapture via try/finally; consolidate triple-open into single seek - Eliminate intermediate clip file: seek directly via CAP_PROP_POS_MSEC - Make MP4 the default output, GIF opt-in via --gif flag - Add return types to all functions - Add Args/Returns docstrings - Use descriptive variable names throughout Made-with: Cursor

…sistency Made-with: Cursor

- Replace Unicode ellipsis and multiplication sign with ASCII equivalents - Fix step numbering from 1-5 to 1-4 (only 4 actual steps) - Move frame_width reading into convert_mp4_to_gif - Remove unused text_height variable Made-with: Cursor

s1lent4gnt

LGTM! Chad

* Add create reward visualization and multimodal analysis tool * add example for creating progress video for sarm * nit * precommit * refactor: address review comments on create_progress_videos.py - Add shebang and Apache 2.0 license header - Replace hardcoded absolute OUTPUT_DIR with relative default (./progress_videos) - Add argparse CLI (--repo-id, --episode, --camera-key, --output-dir, --gif) - Wrap entrypoint in def main() - Replace all print() with logging - Use logging.error/warning instead of traceback.print_exc - Release VideoCapture via try/finally; consolidate triple-open into single seek - Eliminate intermediate clip file: seek directly via CAP_PROP_POS_MSEC - Make MP4 the default output, GIF opt-in via --gif flag - Add return types to all functions - Add Args/Returns docstrings - Use descriptive variable names throughout Made-with: Cursor * refactor: move create_progress_videos.py to examples/dataset/ for consistency Made-with: Cursor * refactor: address PR review comments on create_progress_videos.py - Replace Unicode ellipsis and multiplication sign with ASCII equivalents - Fix step numbering from 1-5 to 1-4 (only 4 actual steps) - Move frame_width reading into convert_mp4_to_gif - Remove unused text_height variable Made-with: Cursor

Add create reward visualization and multimodal analysis tool

0638590

github-actions Bot added the examples Issues related to the examples label Mar 13, 2026

pkooij added 2 commits March 18, 2026 10:13

add example for creating progress video for sarm

219c08c

nit

dad6610

pkooij changed the title ~~Add create reward visualization and multimodal analysis tool~~ Add create reward visualization Mar 18, 2026

pkooij and others added 2 commits March 18, 2026 10:16

Merge branch 'main' into feat/analysis_dataset

c3a6ddb

precommit

e69be57

pkooij self-assigned this Mar 18, 2026

Merge branch 'feat/analysis_dataset' of https://github.com/huggingfac…

46b97da

…e/lerobot into feat/analysis_dataset