diff --git a/lerobot/common/datasets/factory.py b/lerobot/common/datasets/factory.py index d9ce7a0f..26bc45cd 100644 --- a/lerobot/common/datasets/factory.py +++ b/lerobot/common/datasets/factory.py @@ -82,6 +82,7 @@ def make_dataset(cfg, split: str = "train") -> LeRobotDataset | MultiLeRobotData v2.RandomAdjustSharpness( cfg.image_transform.sharpness_factor, p=cfg.image_transform.sharpness_p ), + # Using RandomAdjustSharpness with parameter < 1 adds blur to the image v2.RandomAdjustSharpness(cfg.image_transform.blur_factor, p=cfg.image_transform.blur_p), v2.ToDtype(torch.float32, scale=True), ] diff --git a/lerobot/configs/default.yaml b/lerobot/configs/default.yaml index ebabf5e0..cc707b4b 100644 --- a/lerobot/configs/default.yaml +++ b/lerobot/configs/default.yaml @@ -61,8 +61,13 @@ wandb: image_transform: enable: false colorjittor_factor: 0.5 - colorjittor_p: 0.5 - sharpness_factor: 2 - sharpness_p: 0.5 + colorjittor_p: 0.5 + # Probability that ColorJittor is applied + sharpness_factor: 2 + # Should be more than 1, setting parameter to 1 does not change the image + sharpness_p: 0.5 + # Probability that Sharpness is applied blur_factor: 0.5 - blur_p: 0.5 + # Should be less than 1, setting parameter to 1 does not change the image + blur_p: 0.5 + # Probability that Blur is applied