From c7c0f292c6a250335743014c6b4323a998a0e987 Mon Sep 17 00:00:00 2001 From: Andras Schmelczer Date: Thu, 27 Jun 2024 22:26:40 +0100 Subject: [PATCH] Add ratio_of_unchanged_images --- src/operations/random_edit.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/operations/random_edit.py b/src/operations/random_edit.py index 3b5cf49..adc4de2 100644 --- a/src/operations/random_edit.py +++ b/src/operations/random_edit.py @@ -14,6 +14,11 @@ import numpy as np def random_edit(img: Image, seed: int = 42) -> Image: np.random.seed(seed) + + ratio_of_unchanged_images = 0.95 + if np.random.uniform(0, 1) > ratio_of_unchanged_images: + return img + img = img.convert("RGB") img = ImageEnhance.Contrast(img).enhance(random(0.5, 1.5))