Commit Graph

597 Commits

Author SHA1 Message Date
Remi Cadene ca81b1d6f4 WIP 2024-06-08 12:10:04 +02:00
Simon Alibert e52942a200 fix TestMakeImageTransforms 2024-06-07 17:23:54 +02:00
Simon Alibert b60810a8b6 Update config doc 2024-06-07 10:55:18 +00:00
Simon Alibert faacb36271 transform -> image_transforms 2024-06-06 16:53:37 +00:00
Simon Alibert c45dd8f848 rename to image_transforms 2024-06-06 16:50:22 +00:00
Simon Alibert a86f387554 WIP 2024-06-06 15:24:31 +00:00
Marina Barannikov bdc0ebd36a Updated default transform parameters 2024-06-06 13:50:48 +00:00
Marina Barannikov 19f4a6568d Updated default image_transform parameters 2024-06-06 13:37:58 +00:00
Marina Barannikov 9552a4f010 Added clarification comments 2024-06-06 09:24:58 +00:00
Marina Barannikov d657139828 Updated comments 2024-06-06 09:23:39 +00:00
Simon Alibert b1714803a3 Disable image_transform by default 2024-06-06 08:39:52 +00:00
Simon Alibert 5d55b19cbd Fix tests 2024-06-05 16:47:52 +00:00
Simon Alibert 641d349df4 Add save_image_transforms.py & artifacts 2024-06-05 16:30:47 +00:00
Simon Alibert e444b0d529 Add first tests 2024-06-05 16:29:54 +00:00
Marina Barannikov 8237ed9aa4 Updated visualize script 2024-06-05 16:01:37 +00:00
Simon Alibert 82e32f1fcd Fix RandomSubsetApply weighted sampling 2024-06-05 14:19:37 +00:00
Marina Barannikov 644e77e413 Renamed scripts 2024-06-05 13:35:41 +00:00
Marina Barannikov 1b1bbb1632 Minor formatting 2024-06-05 13:31:40 +00:00
Marina Barannikov 0fb3dd745b Implented visualize_image_transforms script 2024-06-05 13:30:32 +00:00
Marina Barannikov 4dbc1adb0d Updated show_transform to match config 2024-06-05 12:32:53 +00:00
Simon Alibert ceb95592af Remove prints 2024-06-05 12:21:00 +00:00
Simon Alibert 6509c3f6d4 Implement RandomSubsetApply features 2024-06-05 12:15:36 +00:00
Marina Barannikov 8b134725d5
Merge branch 'huggingface:main' into 2024_05_30_add_data_augmentation 2024-06-05 13:56:47 +02:00
Marina Barannikov a544949ebe Added example of torchvision image augmentation on LeRobotDataset 2024-06-05 10:53:18 +00:00
Simon Alibert fdf56e7a62 Redesign config 2024-06-05 09:49:31 +00:00
Simon Alibert 443b06b412 refactor show_image_transforms 2024-06-05 09:34:39 +00:00
Alexander Soare 1eb4bfe2e4
Fix videos_dir documentation (#247) 2024-06-05 08:25:20 +01:00
Alexander Soare 21f222fa1d
Add out_dir option to eval (#244) 2024-06-04 21:01:53 +02:00
amandip7 33362dbd17
Adding parameter dataloading_s to console logs and wandb for tracking… (#243)
Co-authored-by: Remi <re.cadene@gmail.com>
2024-06-04 17:02:05 +01:00
Marina Barannikov 22bd1f0669 Updated formatting 2024-06-04 12:06:36 +00:00
Marina Barannikov 31e3c82386 Merge remote-tracking branch 'refs/remotes/origin/2024_05_30_add_data_augmentation' into 2024_05_30_add_data_augmentation 2024-06-04 12:00:46 +00:00
Marina Barannikov 5eea2542d9 Added visualisations for image augmentation 2024-06-04 11:57:45 +00:00
Marina Barannikov 42f9cc9c2a Updated transforms arguments 2024-06-04 11:14:54 +00:00
Marina Barannikov 66629a956d Updated config to match transforms 2024-06-04 11:09:23 +00:00
Simon Alibert 7be2c35c0a
Merge branch 'huggingface:main' into 2024_05_30_add_data_augmentation 2024-06-04 12:28:06 +02:00
Ruijie b0d954c6e1
Fix bug in normalize to avoid divide by zero (#239)
Co-authored-by: rj <rj@teleopstrio-razer.lan>
Co-authored-by: Remi <re.cadene@gmail.com>
2024-06-04 12:21:28 +02:00
Marina Barannikov 14291171cc Updated default.yaml 2024-06-03 17:21:16 +00:00
Marina Barannikov cc4b3bd8e7 Updated default.yaml 2024-06-03 17:18:33 +00:00
Simon Alibert 602ea9844b Add RandomSubsetApply 2024-06-03 17:15:37 +00:00
Simon Alibert bd3111f28b
Fix `visualize_dataset.py --help` (#241) 2024-06-03 16:35:16 +02:00
Marina Barannikov 9f8415fa83 Added clarification comments 2024-06-03 14:18:08 +00:00
Alexander Soare cf15cba5fc
Remove redundant slicing operation in Diffusion Policy (#240) 2024-06-03 13:04:24 +01:00
jganitzer 042e193995
Typo in examples\4_train_policy_with_script.md (#235) 2024-05-31 18:14:14 +01:00
Marina Barannikov 212a5ab29b Updated implementation on MultiLeRobotDataset 2024-05-31 16:28:08 +00:00
Marina Barannikov c4870e5892 Added data augmentation feature to MultiLeRobotDataset 2024-05-31 15:42:31 +00:00
Marina Barannikov 20a3715469 Merge remote-tracking branch 'origin/main' into 2024_05_30_add_data_augmentation 2024-05-31 14:50:31 +00:00
marina.barannikov@huggingface.co 65e46a49e1 Implemented data augmentation with LeRobot class 2024-05-31 14:16:38 +00: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
Radek Osmulski 83f4f7f7e8
Add precision param to format_big_number (#232) 2024-05-31 10:19:01 +02:00