Commit Graph

614 Commits

Author SHA1 Message Date
Remi Cadene
7a659dbd6b Add reset-time-s, Add keyboard early exit, Add comments 2024-07-12 12:58:56 +02:00
Remi Cadene
1993d29296 Remove int32 convertion function, Set PID 1500, 0, 400 for elbow 2024-07-12 11:24:29 +02:00
Remi Cadene
e615176845 nit 2024-07-11 20:03:56 +02:00
Remi Cadene
3918e118c3 Add assert_same_address 2024-07-11 18:38:23 +02:00
Remi Cadene
17bc32c1e3 nit 2024-07-11 17:16:24 +02:00
Remi Cadene
73692e7459 Make data recording smooth and small fix 2024-07-11 17:14:23 +02:00
Remi Cadene
e6aa8cc641 more ports 2024-07-11 15:27:03 +02:00
Remi Cadene
5494faf096 fix unit tests (minimal) 2024-07-11 00:24:58 +02:00
Remi Cadene
01f8cede0b Add koch to CI 2024-07-10 23:05:36 +02:00
Remi Cadene
5d092b1be2 nit 2024-07-10 20:01:29 +02:00
Remi Cadene
7560372404 removed to radian representations 2024-07-10 19:56:51 +02:00
Remi Cadene
1bc13f39bf rename color to color_mode 2024-07-10 19:54:23 +02:00
Remi Cadene
95de4b7454 Skip tests if alexander koch robot is not available 2024-07-10 19:41:57 +02:00
Remi Cadene
c2388c59be uncomment shutil.rmtree(tmp_imgs_dir) 2024-07-10 19:07:15 +02:00
Remi Cadene
f7a7ed9aa9 Update opencv and dynamixel (find_port) 2024-07-10 18:47:59 +02:00
Remi Cadene
7411cf2a7a Improve opencv 2024-07-10 17:43:33 +02:00
Remi Cadene
2bebdf78a0 Nit + Remove benchmark opencv + Clean + Add docstring 2024-07-10 17:35:18 +02:00
Remi Cadene
fb06417a83 remove comment read/write single value 2024-07-10 14:13:22 +02:00
Remi Cadene
dd7bce7563 format 2024-07-10 14:11:53 +02:00
Remi Cadene
68a561570c fix unit test 2024-07-10 14:05:58 +02:00
Remi Cadene
52e760a88e Almost done 2024-07-10 00:07:40 +02:00
Remi Cadene
798373e7bf All tests passing except test_control_robot.py 2024-07-09 22:57:36 +02:00
Remi Cadene
a0432f1608 Add unit tests (WIP) 2024-07-09 22:57:36 +02:00
Remi Cadene
3ba05d53b9 Remove async in dynamixel 2024-07-09 22:57:36 +02:00
Remi Cadene
72b2e342ae revert threads to thread, results to color_image in OpenCVCamera 2024-07-09 22:57:36 +02:00
Remi Cadene
d83d34d9b3 Add log_control_info 2024-07-09 22:57:36 +02:00
Remi Cadene
3ff789c181 Add async_read and async_write (it doesnt work) 2024-07-09 22:57:36 +02:00
Remi Cadene
6e77a399a2 Add dynamixel-sdk as extra 2024-07-09 22:57:35 +02:00
Remi Cadene
8a7aa50e97 Style 2024-07-09 22:57:12 +02:00
Remi Cadene
47aac0dff7 rebase lock 2024-07-09 22:57:11 +02:00
Remi Cadene
858d49fc04 Add robot_devices and control_robot script 2024-07-09 22:56:46 +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
cc2f6e7404 Train diffusion pusht_keypoints (#307)
Co-authored-by: Remi <re.cadene@gmail.com>
2024-07-09 12:35:50 +01: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
Alexander Soare
7bd5ab16d1 Fix generation of dataset test artifact (#306) 2024-07-05 11:02:26 +01:00
Simon Alibert
74362ac453 Add VQ-BeT copyrights (#299) 2024-07-04 13:02:31 +02:00
Simon Alibert
964f9e86d6 Cleanup config defaults (#300) 2024-07-04 11:53:29 +02:00
Nur Muhammad "Mahi" Shafiullah
7a5fc76b9f Added new credits and citations (#301) 2024-07-03 20:57:47 +02:00
Alexander Soare
342f429f1c Add test to make sure policy dataclass configs match yaml configs (#292) 2024-06-26 09:09:40 +01:00
Seungjae Lee
7d1542cae1 Add VQ-BeT (#166) 2024-06-26 08:55:02 +01:00
Alexander Soare
9aa4cdb976 Checkpoint on final step of training even when it doesn't coincide with save_freq. (#284) 2024-06-20 08:27:01 +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
48951662f2 Bug fix: missing attention mask in VAE encoder in ACT policy (#279)
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
2024-06-19 12:07:21 +01:00
Thomas Wolf
56199fb76f Update readme to detail the lerobot dataset format (#275)
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
2024-06-18 13:40:03 +01:00
Thomas Wolf
11f1cb5dc9 Bug fix: fix setting different learning rates between backbone and main model in ACT policy (#280) 2024-06-18 13:31:35 +01:00
Jihoon Oh
b72d574891 fix Unet global_cond_dim to use state dim, not action dim (#278) 2024-06-17 15:17:28 +01:00
Alexander Soare
15dd682714 Add multi-image support to diffusion policy (#218) 2024-06-17 08:11:20 +01:00
Marina Barannikov
e28fa2344c added visualization for min and max transforms (#271)
Co-authored-by: Simon Alibert <alibert.sim@gmail.com>
2024-06-17 09:09:57 +02:00
Simon Alibert
a92d79fff2 Fix nightlies (#273) 2024-06-14 17:11:19 +01: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