Commit Graph

778 Commits

Author SHA1 Message Date
Simon Alibert d0d8193625 Add unitreeh and umi 2024-10-17 23:33:51 +02:00
Alexander Soare cd0fc261c0
Make `say(blocking=True)` work for Linux (#460) 2024-10-17 15:22:21 +01:00
Simon Alibert 7242c57400 Cleanup 2024-10-17 16:08:37 +02:00
Simon Alibert 3ee3739edc Add batch conversion log 2024-10-17 13:08:58 +02:00
Simon Alibert ad3f112d16 Add fixes for lfs tracking 2024-10-17 12:58:48 +02:00
Simon Alibert 50a75ad3fe Write episodes as jsonlines 2024-10-17 10:17:27 +02:00
Simon Alibert c146ba936f Add episode chunks logic, move_videos & lfs tracking fix 2024-10-16 23:34:54 +02:00
Remi 77478d50e5
Refactor `record` with `add_frame` (#468)
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
2024-10-16 20:51:35 +02:00
Simon Alibert 110264000f Add fixes for batch convert 2024-10-15 19:03:11 +02:00
Simon Alibert 9433ac52ec WIP add batch convert 2024-10-15 13:08:31 +02:00
Simon Alibert da78bbfd16 Update load_tasks doc 2024-10-15 11:06:28 +02:00
Simon Alibert 835ab5a81b Cleanup, fix load_tasks 2024-10-15 11:05:16 +02:00
Simon Alibert f96773de10 Fix safe_version 2024-10-14 13:51:40 +02:00
Simon Alibert cbc51e1341 Extend v1 compatibility 2024-10-14 10:14:27 +02:00
Simon Alibert cf633344be Add multitask support, refactor conversion script 2024-10-13 21:21:40 +02:00
Simon Alibert 8bd406e607 Add suggestions from code review 2024-10-11 18:52:11 +02:00
Simon Alibert 3ea53124e0 Add padding keys and download_data option 2024-10-11 17:38:47 +02:00
Simon Alibert 7f680886b0 Add huggingface-hub patch for offline snapshot_download with local_dir 2024-10-11 11:03:11 +02:00
Simon Alibert 6d2bc11365 Add doc, scrap video_frame_keys attribute 2024-10-11 10:59:38 +02:00
Simon Alibert b417cebc4e Update LeRobotDataset.__get_item__ 2024-10-10 21:32:14 +02:00
Remi 97b1feb0b3
Add policy/act_aloha_real.yaml + env/act_real.yaml (#429)
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
2024-10-10 17:12:45 +02:00
Simon Alibert 3113038beb Merge remote-tracking branch 'origin/main' into user/aliberts/2024_09_25_reshape_dataset 2024-10-09 14:33:58 +02:00
Simon Alibert 096824b5ff Rework LeRobotDataset.__init__ 2024-10-09 14:33:26 +02:00
Eugene Mironov c29e70e5a1
Fix issue with wrong using index instead of camera_index in opencv (#466) 2024-10-09 11:35:19 +02:00
Simon Alibert 2d75b93ba0 Update info.json format 2024-10-08 15:31:37 +02:00
Remi d5b669634a
Fix nightly by updating .cache in dockerignore (#464) 2024-10-07 11:35:35 +02:00
Simon Alibert 21ba4b5263 Add pixel channels 2024-10-06 11:16:49 +02:00
Simon Alibert 028c17fd48 Merge remote-tracking branch 'origin/main' into user/aliberts/2024_09_25_reshape_dataset 2024-10-04 18:59:41 +02:00
Simon Alibert 1a343c3591
Add support for Stretch (hello-robot) (#409)
Co-authored-by: Remi <remi.cadene@huggingface.co>
Co-authored-by: Remi Cadene <re.cadene@gmail.com>
2024-10-04 18:56:42 +02:00
Simon Alibert 07e113ce21 Add info.json link 2024-10-04 14:36:11 +02:00
Simon Alibert 1016a983a1 Add upload folders 2024-10-04 14:26:50 +02:00
Simon Alibert 17a1214e25 Merge remote-tracking branch 'origin/main' into user/aliberts/2024_09_25_reshape_dataset 2024-10-04 11:22:22 +02:00
Simon Alibert ad115b6c27 WIP 2024-10-03 20:00:44 +02:00
Remi 26f97cfd17
Enable CI for robot devices with mocked versions (#398)
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
2024-10-03 17:05:23 +02:00
Simon Alibert 72f402d44b
Fix dataset card (#453) 2024-09-25 16:56:05 +02:00
Alexander Soare 92573486a8
Don't use async envs by default (#448) 2024-09-20 15:22:52 +02:00
Simon Alibert c712d68f6a
Fix nightlies (#443) 2024-09-18 14:51:45 +02:00
Dana Aubakirova f431a08efa
small fix: assertion error message in envs/utils.py (#426)
Co-authored-by: Remi <re.cadene@gmail.com>
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
Co-authored-by: Remi <remi.cadene@huggingface.co>
2024-09-12 18:03:34 +02:00
Remi beaa427504
Fix slow camera fps with Aloha (#433) 2024-09-12 14:20:24 +02:00
Mishig a88dd602d9
[Vizualization] Better error message (#430)
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
2024-09-12 10:46:48 +02:00
Mishig 6c0324f467
[Vizualization] Fix video layout (#431) 2024-09-12 10:06:29 +02:00
Alexander Soare a60d27b132
Raise ValueError if horizon is incompatible with downsampling (#422) 2024-09-09 17:22:46 +01:00
Mishig 9c463661c1
[Vizualization] Better UI on small screens (like in smartphones) (#423) 2024-09-09 15:39:40 +02:00
Mishig 4255655618
[Vizualization] Show user error if videos codec is not supported (#424) 2024-09-09 15:38:41 +02:00
Joe Clinton f17d9a2ba1
Bug: Fix VQ-Bet not working when n_action_pred_token=1 (#420)
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
2024-09-09 09:41:13 +01:00
Remi 9ff829a3a1
Add comments for Aloha (#417)
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
2024-09-06 21:07:52 +02:00
Mishig d6516f0e03
[Visualization tool] Fix when dim state != dim action (#415) 2024-09-06 17:07:26 +02:00
Jack Vial b0b8612eff
fix(calibrate): fix calibrate arms option type. should be str not int (#418)
Co-authored-by: Remi <remi.cadene@huggingface.co>
2024-09-06 14:44:31 +02:00
Mishig 1072a055db
[Visualization tool] Fix videos sync (#416) 2024-09-06 10:16:08 +02:00
Remi 9c9f5cac90
Add IntelRealSenseCamera (#410)
Co-authored-by: Simon Alibert <simon.alibert@huggingface.co>
Co-authored-by: shantanuparab-tr <shantanu@trossenrobotics.com>
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
2024-09-05 23:59:41 +02:00