Update transforms (increase sharpness, reduce saturation, reduce hue)

2024-07-16 18:04:55 +02:00 · 2024-07-16 18:04:55 +02:00 · 6fc1d0dfc1
parent 694a5dbca8
commit 6fc1d0dfc1
2 changed files with 4 additions and 4 deletions
--- a/lerobot/configs/default.yaml
+++ b/lerobot/configs/default.yaml
@ -72,13 +72,13 @@ training:
      min_max: [0.8, 1.2]
    saturation:
      weight: 1
-      min_max: [0.5, 1.5]
+      min_max: [0.7, 1.3]
    hue:
      weight: 1
-      min_max: [-0.05, 0.05]
+      min_max: [-0.02, 0.02]
    sharpness:
      weight: 1
-      min_max: [0.8, 1.2]
+      min_max: [0.1, 1.9]

 eval:
  n_episodes: 1
--- a/lerobot/scripts/visualize_image_transforms.py
+++ b/lerobot/scripts/visualize_image_transforms.py
@ -172,4 +172,4 @@ def visualize_transforms_cli(cfg):


 if __name__ == "__main__":
-    visualize_transforms()
+    visualize_transforms_cli()