Implemented data augmentation with LeRobot class

2024-05-31 14:16:38 +00:00
parent 265b0ec44d
commit 65e46a49e1
3 changed files with 20 additions and 2 deletions
--- a/lerobot/configs/default.yaml
+++ b/lerobot/configs/default.yaml
@@ -53,3 +53,12 @@ wandb:
  disable_artifact: false
  project: lerobot
  notes: ""
+
+image_transform:
+  enable: false 
+  colorjittor_factor: 0.5
+  colorjittor_p: 0.5
+  sharpness_factor: 2
+  sharpness_p: 0.5
+  blur_factor: 0.5
+  blur_p: 0.5