Commit Graph

744 Commits

Author SHA1 Message Date
Simon Alibert 299451af81 Add add_episode & task logic 2024-10-21 19:30:20 +02:00
Simon Alibert 9ebf8b88ec Merge remote-tracking branch 'origin/main' into user/aliberts/2024_09_25_reshape_dataset 2024-10-21 00:22:30 +02:00
Simon Alibert c1232a01e2 Add add_frame, empty dataset creation 2024-10-21 00:16:52 +02:00
Simon Alibert 3b925c3dce Add ImageWriter 2024-10-21 00:15:09 +02:00
Simon Alibert e46bdb9d30 Change card creation 2024-10-20 14:01:10 +02:00
Simon Alibert 9316cf46ef Add file paths 2024-10-20 14:00:19 +02:00
Simon Alibert ac3798bd62 Move default paths, use jsonlines for tasks 2024-10-18 17:53:25 +02:00
Simon Alibert bce3dc3bfa Add load_metadata 2024-10-18 14:59:09 +02:00
Simon Alibert 1a51505ec6 Add download_metadata, move default paths 2024-10-18 14:48:34 +02:00
Simon Alibert e7355ba595 Fix episodes.jsonl 2024-10-18 12:03:29 +02:00
Boris Zimka c351e1fff9
Fix gymnasium version as pre-1.0.0 (#471)
Co-authored-by: Remi <re.cadene@gmail.com>
Co-authored-by: Remi <remi.cadene@huggingface.co>
2024-10-18 10:23:27 +02:00
Simon Alibert 91e8ce772b Remove caret requirement 2024-10-18 09:48:38 +02:00
Simon Alibert 3a9f964429 Add copyrights 2024-10-18 00:31:21 +02:00
Simon Alibert beacb7e957 Cleanup 2024-10-18 00:30:16 +02:00
Simon Alibert be64d54bd9 Update doc 2024-10-18 00:29:50 +02:00
Simon Alibert 354f37aed3 Merge remote-tracking branch 'origin/main' into user/aliberts/2024_09_25_reshape_dataset 2024-10-17 23:34:00 +02:00
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