45a03d253a
[pre-commit.ci] auto fixes from pre-commit.com hooks
...
for more information, see https://pre-commit.ci
2025-03-24 09:40:36 +00:00
dc91068712
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-03-24 09:22:38 +00:00
4c4a356179
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-03-24 09:22:38 +00:00
8f97f58ba5
Added possiblity to record and replay delta actions during teleoperation rather than absolute actions
...
Co-authored-by: Adil Zouitine <adilzouitinegm@gmail.com>
2025-03-24 09:22:38 +00:00
c504d23f82
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-03-24 09:13:12 +00:00
97b8214cd2
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>
2025-03-24 09:04:12 +00:00
c4c2ce04e7
Update pre-commits ( #733 )
2025-02-15 15:51:17 +01:00
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
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
1ee1acf8ad
Comply with torchvision 0.21 custom transforms ( #665 )
2025-01-30 22:06:11 +01:00
c4d912a241
Check for "/" in feature names ( #660 )
2025-01-29 21:54:49 +01:00
b8b368310c
typo fix: batch_convert_dataset_v1_to_v2.py ( #615 )
...
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
2025-01-09 09:57:45 +01:00
59e2757434
Fix broken `create_lerobot_dataset_card` ( #590 )
2024-12-23 15:05:59 +01:00
73b64c3089
[vizualizer] for LeRobodDataset V2 ( #576 )
2024-12-20 16:26:23 +01:00
4c41f6fcc6
Fix example 6 ( #572 )
2024-12-11 10:32:18 +01:00
44f9b21e74
LerobotDataset pushable to HF from any folder ( #563 )
2024-12-09 11:32:25 +01:00
a2c181992a
Refactor OpenX ( #505 )
2024-12-03 00:51:55 +01:00
32eb0cec8f
Dataset v2.0 ( #461 )
...
Co-authored-by: Remi <remi.cadene@huggingface.co>
2024-11-29 19:04:00 +01:00
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
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
72f402d44b
Fix dataset card ( #453 )
2024-09-25 16:56:05 +02:00
9d0c6fe419
Fix nightlies & untrack json files from git lfs ( #414 )
2024-09-05 15:07:43 +02:00
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
b98ea415c1
Add dataset cards ( #363 )
2024-08-16 10:08:44 +02:00
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
f8a6574698
Add online training with TD-MPC as proof of concept ( #338 )
2024-07-25 11:16:38 +01:00
0b21210d72
Convert datasets to av1 encoding ( #302 )
2024-07-22 20:08:59 +02:00
8865e19c12
Fix datasets missing versions ( #318 )
2024-07-16 23:02:31 +02:00
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
5ffcb48a9a
Add available list of raw repo ids ( #312 )
2024-07-13 11:30:50 +02:00
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
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
2abef3bef9
Enable `video_reader` backend ( #220 )
...
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
2024-06-19 17:15:25 +02:00
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
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
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
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
504d2aaf48
add EpisodeAwareSampler ( #217 )
...
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
2024-05-31 13:43:47 +01:00
633115d861
Fix chaining in MultiLerobotDataset ( #233 )
2024-05-31 09:03:28 +01:00
111cd58f8a
Add `MultiLerobotDataset` for training with multiple `LeRobotDataset`s ( #229 )
2024-05-30 16:12:21 +01:00
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
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
e3b9f1c19b
Add resume training ( #205 )
...
Co-authored-by: Remi <re.cadene@gmail.com>
2024-05-28 12:04:23 +01:00
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
9b62c25f6c
Adds split_by_episodes to LeRobotDataset ( #158 )
2024-05-20 14:04:04 +02:00
01eae09ba6
Fix aloha real-world datasets ( #175 )
2024-05-20 13:48:09 +02:00
f52f4f2cd2
Add copyrights ( #157 )
2024-05-15 12:13:09 +02:00
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
6eaffbef1d
Add 25 real world datasets (static + mobile aloha) ( #133 )
2024-05-05 19:13:11 +02:00
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
pre-commit-ci[bot]
Michel Aractingi
Yoel
Simon Alibert
Remi
Thomas Lips
CharlesCNorton
Eugene Mironov
Mishig
Claudio Coppola
Remi
Alexander Soare
Thomas Wolf
Marina Barannikov
Radek Osmulski