Skip to content

Commit

Permalink
Neatify
Browse files Browse the repository at this point in the history
  • Loading branch information
marinabar committed Jun 13, 2024
1 parent d15081e commit 1760045
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions lerobot/scripts/visualize_image_transforms.py
Original file line number Diff line number Diff line change
Expand Up @@ -127,7 +127,7 @@ def save_config_single_transforms(cfg, original_frame, output_dir):
}
tf = get_image_transforms(**kwargs)
transformed_frame = tf(original_frame)
to_pil(transformed_frame).save(output_dir_single / f"min.png", quality=100)
to_pil(transformed_frame).save(output_dir_single / "min.png", quality=100)

# Apply max transformation
kwargs = {
Expand All @@ -136,7 +136,7 @@ def save_config_single_transforms(cfg, original_frame, output_dir):
}
tf = get_image_transforms(**kwargs)
transformed_frame = tf(original_frame)
to_pil(transformed_frame).save(output_dir_single / f"max.png", quality=100)
to_pil(transformed_frame).save(output_dir_single / "max.png", quality=100)

print(f" {output_dir_single}")

Expand Down

0 comments on commit 1760045

Please sign in to comment.