tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Added visualisations for image augmentation

Browse Source
This commit is contained in:
Marina Barannikov
2024-06-04 11:57:45 +00:00
parent 42f9cc9c2a
commit 5eea2542d9
3 changed files with 74 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lerobot/configs/default.yaml
View File

@@ -60,10 +60,9 @@ wandb:
image_transform:
enable: false
colorjittor_range: (0, 1)
colorjittor_p: 1
# Range from which to sample colorjittor factor
sharpness_factor: 3
colorjitter_factor: 0.5
colorjitter_p: 1.O
sharpness_factor: 3.0
# Should be more than 1, setting parameter to 1 does not change the image
sharpness_p: 0.5
blur_factor: 0.2
@@ -71,4 +70,4 @@ image_transform:
blur_p: 0.5
n_subset: 3
# Maximum number of transforms to apply
list: ["colorjittor", "sharpness", "blur"]
list: ["colorjitter", "sharpness", "blur"]