ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-04 21:01:26 +00:00
Code Issues Packages Projects Releases Wiki Activity
757 Commits 208 Branches 9 Tags
ff82367c628548ffcc52a747f43ecf2e8a42e487
Commit Graph

166 Commits

Author SHA1 Message Date
Michel Aractingi
36711d766a Modified crop_dataset_roi interface to automatically write the cropped parameters to a json file in the meta of the dataset 
Co-authored-by: Adil Zouitine <adilzouitinegm@gmail.com>
 2025-02-14 12:32:45 +01:00
Michel Aractingi
c462a478c7 Hardcoded some normalization parameters. TODO refactor 
Added masking actions on the level of the intervention actions and offline dataset

Co-authored-by: Adil Zouitine <adilzouitinegm@gmail.com>
 2025-02-13 14:27:14 +01:00
Michel Aractingi
b9217b06db Added possiblity to record and replay delta actions during teleoperation rather than absolute actions 
Co-authored-by: Adil Zouitine <adilzouitinegm@gmail.com>
 2025-02-12 19:25:41 +01:00
Yoel
f1c8bfe01e [Port HIL-SERL] Add HF vision encoder option in SAC (#651) 
Added support with custom pretrained vision encoder to the modeling sac implementation. Great job @ChorntonYoel !
 2025-02-03 15:07:58 +00:00
Michel Aractingi
367dfe51c6 Added support for checkpointing the policy. We can save and load the policy state dict, optimizers state, optimization step and interaction step 
Added functions for converting the replay buffer from and to LeRobotDataset. When we want to save the replay buffer, we convert it first to LeRobotDataset format and save it locally and vice-versa.

Co-authored-by: Adil Zouitine <adilzouitinegm@gmail.com>
 2025-02-03 15:07:58 +00:00
CharlesCNorton
c2f7af3339 typo fix: batch_convert_dataset_v1_to_v2.py (#615) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2025-02-03 15:04:03 +00:00
Eugene Mironov
4a43c83522 Fix broken create_lerobot_dataset_card (#590) 2025-02-03 15:04:02 +00:00
Mishig
0a4e9e25d0 [vizualizer] for LeRobodDataset V2 (#576) 2025-02-03 15:04:02 +00:00
Yoel
e35546f58e Reward classifier and training (#528) 
Co-authored-by: Daniel Ritchie <daniel@brainwavecollective.ai>
Co-authored-by: resolver101757 <kelster101757@hotmail.com>
Co-authored-by: Jannik Grothusen <56967823+J4nn1K@users.noreply.github.com>
Co-authored-by: Remi <re.cadene@gmail.com>
Co-authored-by: Michel Aractingi <michel.aractingi@huggingface.co>
 2024-12-17 02:41:29 +07:00
Simon Alibert
4c41f6fcc6 Fix example 6 (#572) 2024-12-11 10:32:18 +01:00
Claudio Coppola
44f9b21e74 LerobotDataset pushable to HF from any folder (#563) 2024-12-09 11:32:25 +01:00
Michel Aractingi
a2c181992a Refactor OpenX (#505) 2024-12-03 00:51:55 +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
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
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
Simon Alibert
9d0c6fe419 Fix nightlies & untrack json files from git lfs (#414) 2024-09-05 15:07:43 +02:00
Michel Aractingi
eb4c505cff Support for converting OpenX datasets from RLDS format to LeRobotDataset (#354) 
Signed-off-by: youliangtan <tan_you_liang@hotmail.com>
Co-authored-by: Simon Alibert <alibert.sim@gmail.com>
Co-authored-by: youliangtan <tan_you_liang@hotmail.com>
Co-authored-by: Remi <re.cadene@gmail.com>
 2024-08-27 09:07:00 +02:00
Simon Alibert
b98ea415c1 Add dataset cards (#363) 2024-08-16 10:08:44 +02:00
Remi
bbe9057225 Improve control robot ; Add process to configure motor indices (#326) 
Co-authored-by: Simon Alibert <alibert.sim@gmail.com>
Co-authored-by: jess-moss <jess.moss@dextrousrobotics.com>
Co-authored-by: Marina Barannikov <marina.barannikov@huggingface.co>
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-08-15 18:11:33 +02:00
Alexander Soare
f8a6574698 Add online training with TD-MPC as proof of concept (#338) 2024-07-25 11:16:38 +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
Remi
5e54e39795 Add real robot devices and scripts to control real robot (#288) 
Co-authored-by: Simon Alibert <alibert.sim@gmail.com>
 2024-07-15 17:43:10 +02:00
Remi
5ffcb48a9a Add available list of raw repo ids (#312) 2024-07-13 11:30:50 +02:00
Simon Alibert
e410e5d711 Improve video benchmark (#282) 
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
Co-authored-by: Remi <re.cadene@gmail.com>
 2024-07-09 20:20:25 +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
Thomas Wolf
125bd93e29 Improve push_dataset_to_hub API + Add unit tests (#231) 
Co-authored-by: Remi <re.cadene@gmail.com>
Co-authored-by: Simon Alibert <alibert.sim@gmail.com>
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2024-06-13 15:18:02 +02:00
Marina Barannikov
c38f535c9f FIx make_dataset to match transforms config (#264) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2024-06-12 19:45:42 +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
Remi
d585c73f9f Add real-world support for ACT on Aloha/Aloha2 (#228) 
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-05-31 15:31:02 +02:00
Radek Osmulski
504d2aaf48 add EpisodeAwareSampler (#217) 
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-05-31 13:43:47 +01: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
Remi
265b0ec44d Refactor env to add key word arguments from config yaml (#223) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2024-05-30 13:45:22 +02:00
Remi
2c2e4e14ed Add aloha_dora_format.py (#201) 
Co-authored-by: Thomas Wolf <thomwolf@users.noreply.github.com>
 2024-05-30 11:26:39 +02:00
Alexander Soare
e3b9f1c19b Add resume training (#205) 
Co-authored-by: Remi <re.cadene@gmail.com>
 2024-05-28 12:04:23 +01:00
Remi
c4da689171 Hot fix to compute validation loss example test (#200) 
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-05-20 18:30:11 +02: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
6eaffbef1d Add 25 real world datasets (static + mobile aloha) (#133) 2024-05-05 19:13:11 +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
Alexander Soare
9d60dce6f3 Tidy up yaml configs (#121) 2024-04-30 16:08:59 +01:00
Remi
e4e739f4f8 Refactor push_dataset_to_hub (#118) 2024-04-30 14:25:41 +02:00
Simon Alibert
2765877f28 Speed up CI, add more checks (#115) 
- Split pytest and end-to-end tests into separate jobs
- Add poetry check to ensure pyproject.toml and poetry.lock are in sync
- Add ruff format --diff to ensure style formatting is applied (fails if ruff would reformat anything)
 2024-04-29 23:05:55 +02:00
Adil Zouitine
1ec5f77f7c Remove mask check key in XarmProcessor (#117) 2024-04-29 22:00:37 +02:00