andras/bipolaroidbipolaroid
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Tune edits

Browse source
This commit is contained in:
Andras Schmelczer 2024-06-18 22:54:55 +01:00
parent e17068baba
commit 388d3d3ce3
No known key found for this signature in database
GPG key ID: FC8F2C3D3D1A718C
3 changed files with 28 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

28
src/editor/training/random_edit.py
View file

@ -1,20 +1,32 @@
from PIL import Image, ImageEnhance
from ..utils import random, get_colour_lut, apply_pixel_shader
from ..operations import add_noise, add_random_colour_spill
from ..utils import random
from ..operations import (
add_noise,
add_random_colour_spill,
get_random_gamma,
adjust_gamma,
apply_pixel_shader,
get_random_brightness_lut,
get_random_saturation_per_hue_lut,
)
import numpy as np
def random_edit(img: Image, seed: int = 42) -> Image:
np.random.seed(seed)
img = add_noise(img, random(0, 0.2))
img = ImageEnhance.Contrast(img).enhance(random(0.5, 2))
img = add_random_colour_spill(img, 1.3)
img = img.convert("RGB")
img = adjust_gamma(img, get_random_gamma())
img = add_noise(img, random(0, 0.1))
img = ImageEnhance.Contrast(img).enhance(random(0.5, 1.5))
img = add_random_colour_spill(img, 0.2)
img = img.convert("HSV")
saturation_lut = get_colour_lut(variance=0.3, count=5, type="linear")
brightness_lut = get_colour_lut(variance=0.3, count=5, type="cubic")
saturation_lut = get_random_saturation_per_hue_lut()
brightness_lut = get_random_brightness_lut()
img = apply_pixel_shader(
img, lambda h, s, v: (h, saturation_lut[s], brightness_lut[v])
img, lambda h, s, v: (h, round(s * saturation_lut[h]), brightness_lut[v])
)
img = img.convert("RGB")
return img