Added clarification comments

2024-06-03 14:18:08 +00:00
parent 212a5ab29b
commit 9f8415fa83
2 changed files with 10 additions and 4 deletions
--- 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),
            ]