wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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

This commit is contained in:
Remi Cadene 2024-07-16 18:04:55 +02:00
parent 694a5dbca8
commit 6fc1d0dfc1
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lerobot/configs/default.yaml
View File

@ -72,13 +72,13 @@ training:
min_max: [0.8, 1.2] min_max: [0.8, 1.2]
saturation: saturation:
weight: 1 weight: 1
min_max: [0.5, 1.5] min_max: [0.7, 1.3]
hue: hue:
weight: 1 weight: 1
min_max: [-0.05, 0.05] min_max: [-0.02, 0.02]
sharpness: sharpness:
weight: 1 weight: 1
min_max: [0.8, 1.2] min_max: [0.1, 1.9]
eval: eval:
n_episodes: 1 n_episodes: 1

2
lerobot/scripts/visualize_image_transforms.py
View File

@ -172,4 +172,4 @@ def visualize_transforms_cli(cfg):
if __name__ == "__main__": if __name__ == "__main__":
visualize_transforms() visualize_transforms_cli()