Allow parametrizing gamma

This commit is contained in:
kris 2021-05-28 11:24:05 +01:00
parent 07a358d49b
commit a593639606

View File

@ -18,8 +18,8 @@ def srgb_to_linear(im: np.ndarray) -> np.ndarray:
return (np.clip(rgb_linear, 0.0, 1.0) * 255).astype(np.float32)
def linear_to_srgb(im: np.ndarray) -> np.ndarray:
srgb = linear_to_srgb_array(im / 255.0, gamma=2.4)
def linear_to_srgb(im: np.ndarray, gamma=2.4) -> np.ndarray:
srgb = linear_to_srgb_array(im / 255.0, gamma=gamma)
return (np.clip(srgb, 0.0, 1.0) * 255).astype(np.float32)