wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
657 Commits 116 Branches 0 Tags 252 MiB
Commit Graph

76 Commits

Author SHA1 Message Date
Remi 07e8716315
Add FeetechMotorsBus, SO-100, Moss-v1 (#419) 
Co-authored-by: jess-moss <jess.moss@huggingface.co>
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2024-10-25 11:23:55 +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
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 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
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 c712d68f6a
Fix nightlies (#443) 2024-09-18 14:51:45 +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
Remi 429a463aff
Control aloha robot natively (#316) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
 2024-09-04 19:28:05 +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
Remi 2252b42337
Add visualize_dataset_html with `http.server` (#188) 2024-08-08 20:19:06 +03:00
Simon Alibert 0b21210d72
Convert datasets to av1 encoding (#302) 2024-07-22 20:08:59 +02:00
Simon Alibert ee200e86cb
Ensure no upper bound constraints on dependencies (#327) 2024-07-18 12:07:15 +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
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
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
Alexander Soare 3d625ae6d3
Handle `crop_shape=None` in Diffusion Policy (#219) 2024-05-28 18:27:33 +01:00
Alexander Soare e3b9f1c19b
Add resume training (#205) 
Co-authored-by: Remi <re.cadene@gmail.com>
 2024-05-28 12:04:23 +01:00
Simon Alibert 35c1ce7a66
Fix install issues (#191) 2024-05-23 14:25:18 +02:00
Remi 01eae09ba6
Fix aloha real-world datasets (#175) 2024-05-20 13:48:09 +02:00
Akshay Kashyap c9069df9f1
Port SpatialSoftmax and remove Robomimic dependency (#182) 
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-05-16 15:34:10 +01:00
Simon Alibert 89c6be84ca
Limit datasets major update (#176) 
Co-authored-by: Quentin Lhoest <42851186+lhoestq@users.noreply.github.com>
Co-authored-by: Quentin Lhoest <lhoest.q@gmail.com>
 2024-05-12 08:15:07 +02:00
Simon Alibert 7bb5b15f4c
Remove dependencies upper bounds constraints (#145) 2024-05-08 17:23:10 +00:00
Simon Alibert eb6bfe01b2
Relax version contraints & poetry lock (#139) 2024-05-06 17:38:29 +02:00
Alexander Soare f3bba0270d
Remove EMA model from Diffusion Policy (#134) 2024-05-05 11:26:12 +01: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
Simon Alibert b7b69fcc3d
Publish environments (#120) 2024-05-03 16:25:45 +02:00
Remi b2cda12f87
Add video decoding to LeRobotDataset (#92) 2024-05-03 00:50:19 +02:00
Alexander Soare a4891095e4
Use PytorchModelHubMixin to save models as safetensors (#125) 
Co-authored-by: Remi <re.cadene@gmail.com>
 2024-05-01 16:17:18 +01:00
Simon Alibert 791506dfb8
Remove warnings (#111) 
- Replace `use_pretrained_backbone` with `pretrained_backbone_weights`
- Bump diffusers' minimum version `0.26.3` -> `0.27.2`
- Add ignore flags in CI's pytest
- Change Box observation spaces in simulation environments
- Set `version_base="1.2"` in Hydra initializations
- Bump einops' minimum version `0.7.0` -> `0.8.0`
 2024-04-29 00:31:33 +02:00
Adil Zouitine 81e490d46f
Add UMI-gripper dataset (#83) 
Co-authored-by: Remi <re.cadene@gmail.com>
 2024-04-28 18:41:07 +02:00
Simon Alibert a4b6c5e3b1
Add torchaudio (#110) 2024-04-27 14:00:38 +02:00
Simon Alibert b980c5dd9e
CI nightlies cpu/gpu & cleanup (#75) 2024-04-25 14:58:39 +02:00
Remi 1030ea0070
Loads episode_data_index and stats during dataset __init__ (#85) 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
 2024-04-23 14:13:25 +02:00
Cadene c6aca7fe44 For Pusht: use hf datasets to train, rename load_data_with_delta_timestamps -> load_previous_and_future_frames 2024-04-16 17:19:06 +00:00
Simon Alibert 70e8de95f7 Clean & update pyproject 2024-04-16 12:09:56 +02:00
Cadene c1a618e567 fix pusht images type from float32 to uint8, update gym-pusht dependencies 2024-04-11 14:29:16 +00:00
Simon Alibert 0593d20348 Fix gym-xarm 2024-04-11 08:51:37 +02:00
Simon Alibert 91ff69d64c Update gym_xarm 2024-04-09 17:08:36 +02:00
Alexander Soare 575891e8ac Merge remote-tracking branch 'upstream/user/rcadene/2024_03_31_remove_torchrl' into refactor_act 2024-04-09 15:19:29 +01:00
Simon Alibert dfaacbcf5a Split dev/test dependencies 2024-04-09 10:40:11 +02:00
Simon Alibert 2573e89e1d Remove direct dependencies 2024-04-09 10:38:08 +02:00
Simon Alibert 274f20b49d Update gym-pusht 2024-04-09 10:25:41 +02:00
Alexander Soare e6c6c2367f Merge remote-tracking branch 'upstream/user/rcadene/2024_03_31_remove_torchrl' into refactor_act 2024-04-09 08:36:28 +01:00
Cadene 19e7661b8d Remove torchrl/tensordict from dependecies + update poetry cpu 2024-04-09 03:50:49 +00:00
Cadene 73dfa3c8e3 tests for tdmpc and diffusion policy are passing 2024-04-09 02:50:32 +00:00
Alexander Soare 50e4c8050c Merge remote-tracking branch 'upstream/user/rcadene/2024_03_31_remove_torchrl' into refactor_act 2024-04-08 17:13:11 +01:00
Simon Alibert 3f6dfa4916 Add gym-aloha, rename simxarm -> xarm, refactor 2024-04-08 16:24:11 +02:00
Alexander Soare 1bab4a1dd5 Eval reproduction works with gym_aloha 2024-04-08 10:23:26 +01:00