hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

Updated transforms arguments

Browse Source
This commit is contained in:
Marina Barannikov
2024-06-04 11:14:54 +00:00
parent 66629a956d
commit 42f9cc9c2a
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lerobot/common/datasets/transforms.py
View File

@@ -47,11 +47,11 @@ class RandomSubsetApply(Transform):
def make_transforms(cfg):
image_transforms = []
if 'jit' in cfg.list_transforms:
if 'jit' in cfg.image_transform.list:
image_transforms.append(v2.ColorJitter(brightness=cfg.colorjitter_range, contrast=cfg.colorjitter_range))
if 'sharpness' in cfg.list_transforms:
if 'sharpness' in cfg.image_transform.list:
image_transforms.append(v2.RandomAdjustSharpness(cfg.sharpness_factor, p=cfg.sharpness_p))
if 'blur' in cfg.list_transforms:
if 'blur' in cfg.image_transform.list:
image_transforms.append(v2.RandomAdjustSharpness(cfg.blur_factor, p=cfg.blur_p))
return v2.Compose(RandomSubsetApply(image_transforms, n_subset=cfg.n_subset), v2.ToDtype(torch.float32, scale=True))