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

193 Commits

Author SHA1 Message Date
Simon Alibert 23f6c875b5 Apply suggestions from code review 2024-11-25 12:44:12 +01:00
Simon Alibert 6203641710 Use HWC for images 2024-11-19 18:48:17 +01:00
Remi 1f13bda25b
Improve dataset v2 (#498) 2024-11-19 12:31:47 +01:00
Simon Alibert c72ad49c9b Skip test_examples 2024-11-18 18:46:00 +01:00
Simon Alibert eda02fade5 Skip test_visualize_local_dataset 2024-11-18 18:06:18 +01:00
Simon Alibert 8546358bc5 Fix test_visualize_dataset_html 2024-11-18 17:54:15 +01:00
Simon Alibert f3630ad910 Fix tests 2024-11-05 19:09:12 +01:00
Simon Alibert 757ea175d3 Fix test_examples 2024-11-03 19:41:50 +01:00
Simon Alibert b69a132737 Fix test_examples 2024-11-03 19:30:56 +01:00
Simon Alibert 7b159a6b22 Move calculate_episode_data_index 2024-11-03 19:13:00 +01:00
Simon Alibert 74270c8c91 Remove reset_episode_index 2024-11-03 19:07:43 +01:00
Simon Alibert 56e4603d5b Deactivate policies backward compatibility test 2024-11-03 18:30:50 +01:00
Simon Alibert e4ba084e25 Add LeRobotDatasetMetadata 2024-11-03 18:07:37 +01:00
Simon Alibert ac79e8cb36 Fix test_online_buffer.py 2024-11-03 13:15:01 +01:00
Simon Alibert df2cb51364 Add todo in skipped test 2024-11-02 20:01:34 +01:00
Simon Alibert 7a342db9c4 Add more options to img factories 2024-11-02 20:01:02 +01:00
Simon Alibert 6b2ec1ed77 Add test_image_writer, accept PIL images, improve ImageWriter perf in main process 2024-11-02 20:00:07 +01:00
Simon Alibert 375abd3020 Add img and img_tensor factories 2024-11-02 13:06:38 +01:00
Simon Alibert 293bdc7f67 Simplify, add test content, add todo 2024-11-01 19:55:28 +01:00
Simon Alibert 2650872b76 Add tasks and episodes factories 2024-11-01 13:37:17 +01:00
Simon Alibert cd1509d805 Mock snapshot_download 2024-11-01 10:58:09 +01:00
Simon Alibert 443a9eec88 Remove/comment obsolete tests 2024-10-31 21:43:29 +01:00
Simon Alibert ab23a4fd27 Add fixtures in test_datasets 2024-10-31 21:32:09 +01:00
Simon Alibert 1267c3e955 Split fixtures into factories and files 2024-10-31 21:29:50 +01:00
Simon Alibert e69f0c5059 Add test_delta_timestamps.py 2024-10-31 13:48:40 +01:00
Simon Alibert ff84024ee9 Add dataset fixtures 2024-10-31 13:46:46 +01:00
Simon Alibert ee51f54cb5 Remove dataset from image_transform tests 2024-10-29 16:08:01 +01:00
Simon Alibert 8d57093ce7 Merge remote-tracking branch 'origin/main' into user/aliberts/2024_09_25_reshape_dataset 2024-10-29 15:25:25 +01:00
Simon Alibert df3d2ec5df Speedup test 2024-10-26 10:59:06 +02:00
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
Simon Alibert 8bcf81fa24 Add todo 2024-10-24 11:38:32 +02:00
Simon Alibert 615894d3fb Add test_same_attributes_defined 2024-10-24 11:37:44 +02:00
Simon Alibert b46db7ea73 Fix tests 2024-10-22 20:14:06 +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
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 9d0c6fe419
Fix nightlies & untrack json files from git lfs (#414) 2024-09-05 15:07:43 +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
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
Alexander Soare 9c7649f140
Make sure `init_hydra_config` does not require any keys (#376) 2024-08-23 12:27:08 +01: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
Alexander Soare f8a6574698
Add online training with TD-MPC as proof of concept (#338) 2024-07-25 11:16:38 +01:00
Alexander Soare abbb1d2367
Make sure policies don't mutate the batch (#323) 2024-07-22 20:38:33 +01:00
Simon Alibert 0b21210d72
Convert datasets to av1 encoding (#302) 2024-07-22 20:08:59 +02:00
Alexander Soare c0101f0948
Fix ACT temporal ensembling (#319) 2024-07-16 10:27:21 +01: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
Alexander Soare 471eab3d7e
Make ACT compatible with "observation.environment_state" (#314) 2024-07-11 13:12:22 +01: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