admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-26 14:09:47 +00:00
Code Issues Packages Projects Releases Wiki Activity
854 Commits 205 Branches 9 Tags
6d9374c785145828386417ca5749824255ba701a
Commit Graph

40 Commits

Author SHA1 Message Date
Remi Cadene 8746276d41 WIP after Francesco discussion 2025-05-28 17:29:41 +02:00
Remi Cadene e07cb52baa In tests: Add use_videos=False by default, Create mp4 file if True, then fix test_datasets and test_aggregate (all passing) 2025-05-12 15:37:02 +02:00
Remi Cadene 0309a9fcbc Speedup data loading 2025-05-06 15:13:50 +00:00
Remi Cadene 588bf96559 Fix aggregate (num_frames, dataset_from_index, index) 2025-05-06 15:13:35 +00:00
Remi Cadene 71715c3914 fix hf_dataset.set_transform(hf_transform_to_torch) 2025-04-23 11:42:21 +02:00
Remi Cadene d518b036d0 Faster self.meta.episodes[...] 
switch back to set_transform instead of set_format

Add video_files_size_in_mb

pre-commit run --all-files
 2025-04-23 09:14:02 +00:00
Remi Cadene 601b5fdbfe Merge remote-tracking branch 'origin/user/rcadene/2025_04_11_dataset_v3' into user/rcadene/2025_04_11_dataset_v3 2025-04-22 08:19:30 +00:00
Remi Cadene 20b74ae1eb fix 2025-04-21 13:38:29 +00:00
Remi Cadene 5bd9cb1e72 Merge remote-tracking branch 'origin/main' into user/rcadene/2025_04_11_dataset_v3 2025-04-21 11:03:12 +02:00
Remi Cadene b0cca75e5e Progress on aggregate_datasets 2025-04-19 19:11:53 +05:30
Remi Cadene eab5543750 Merge (No verify) 2025-04-17 04:46:09 +02:00
Remi Cadene 6b6a990f4c most unit tests passing (TODO: convert datasets) 2025-04-16 21:30:58 +02:00
Remi Cadene c2a05a1fde Fix (Now loading all frames is possible) 2025-04-14 14:47:18 +00:00
Remi Cadene 34c5d4ce07 Most unit tests are passing 2025-04-11 14:04:22 +02:00
Remi Cadene c1b28f0b58 Commit before episodes episodes_stats merging 2025-04-09 15:20:15 +02:00
pre-commit-ci[bot] 437fc29e12 [pre-commit.ci] pre-commit autoupdate (#871) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2025-04-08 06:58:46 +02:00
Remi Cadene 53ecec5fb2 WIP v21 to v30 2025-03-31 07:38:01 +00:00
Steven Palma 1c15bab70f fix(codec): hot-fix for default codec in linux arm platforms (#868) 2025-03-17 13:23:11 +01:00
Jade Choghari 0e98c6ee96 Add torchcodec cpu (#798) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Remi <re.cadene@gmail.com>
Co-authored-by: Remi <remi.cadene@huggingface.co>
Co-authored-by: Simon Alibert <simon.alibert@huggingface.co>
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2025-03-14 16:53:42 +01:00
Simon Alibert e81c36cf74 Fix dataset version tags (#790) 2025-02-28 14:36:20 +01:00
Simon Alibert a1809ad3de Add typos checks (#770) 2025-02-25 23:51:15 +01:00
Simon Alibert 3354d919fc LeRobotDataset v2.1 (#711) 
Co-authored-by: Remi <remi.cadene@huggingface.co>
Co-authored-by: Remi Cadene <re.cadene@gmail.com>
 2025-02-25 15:27:29 +01:00
Remi 638d411cd3 Add Pi0 (#681) 
Co-authored-by: Simon Alibert <simon.alibert@huggingface.co>
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
Co-authored-by: Pablo <pablo.montalvo.leroux@gmail.com>
 2025-02-04 18:01:04 +01:00
Simon Alibert 3c0a209f9f Simplify configs (#550) 
Co-authored-by: Remi <remi.cadene@huggingface.co>
Co-authored-by: HUANG TZU-CHUN <137322177+tc-huang@users.noreply.github.com>
 2025-01-31 13:57:37 +01:00
Thomas Lips c4d912a241 Check for "/" in feature names (#660) 2025-01-29 21:54:49 +01:00
Simon Alibert 32eb0cec8f Dataset v2.0 (#461) 
Co-authored-by: Remi <remi.cadene@huggingface.co>
 2024-11-29 19:04:00 +01:00
Simon Alibert 0b21210d72 Convert datasets to av1 encoding (#302) 2024-07-22 20:08:59 +02:00
Simon Alibert 8865e19c12 Fix datasets missing versions (#318) 2024-07-16 23:02:31 +02:00
Alexander Soare a4d77b99f0 Include observation.environment_state with keypoints in PushT dataset (#303) 
Co-authored-by: Remi <re.cadene@gmail.com>
 2024-07-09 08:27:40 +01:00
Simon Alibert 2abef3bef9 Enable video_reader backend (#220) 
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-06-19 17:15:25 +02:00
Marina Barannikov ff8f6aa6cd Add data augmentation in LeRobotDataset (#234) 
Co-authored-by: Simon Alibert <alibert.sim@gmail.com>
Co-authored-by: Remi Cadene <re.cadene@gmail.com>
 2024-06-11 19:20:55 +02:00
Alexander Soare 633115d861 Fix chaining in MultiLerobotDataset (#233) 2024-05-31 09:03:28 +01:00
Alexander Soare 111cd58f8a Add MultiLerobotDataset for training with multiple LeRobotDatasets (#229) 2024-05-30 16:12:21 +01:00
Radek Osmulski 9b62c25f6c Adds split_by_episodes to LeRobotDataset (#158) 2024-05-20 14:04:04 +02:00
Remi 01eae09ba6 Fix aloha real-world datasets (#175) 2024-05-20 13:48:09 +02:00
Simon Alibert f52f4f2cd2 Add copyrights (#157) 2024-05-15 12:13:09 +02:00
Simon Alibert f5e76393eb Release cleanup (#132) 
Co-authored-by: Kashif Rasul <kashif.rasul@gmail.com>
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
Co-authored-by: Adil Zouitine <adilzouitinegm@gmail.com>
Co-authored-by: Cadene <re.cadene@gmail.com>
 2024-05-06 03:03:14 +02:00
Remi 19812ca470 Add dataset visualization with rerun.io (#131) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2024-05-04 16:07:14 +02:00
Remi b2cda12f87 Add video decoding to LeRobotDataset (#92) 2024-05-03 00:50:19 +02:00
Remi 659c69a1c0 Refactor datasets into LeRobotDataset (#91) 
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-04-25 12:23:12 +02:00