mirror of
https://github.com/kuhyx/WUT_Computer_Science.git
synced 2026-07-04 18:03:14 +02:00
41 lines
849 B
Python
41 lines
849 B
Python
|
from PIL import Image, ImageFilter
|
||
|
|
import numpy as np
|
||
|
|
from pathlib import Path
|
||
|
|
from .colour_functions import sRGB_to_sRGB_linear
|
||
|
|
|
||
|
|
|
||
|
|
def load_image(path):
|
||
|
|
img = Image.open(Path(path))
|
||
|
|
return np.asarray(img)/256.
|
||
|
|
|
||
|
|
|
||
|
|
def load_image_with_blur(path, blur = 0.):
|
||
|
|
img = Image.open(Path(path))
|
||
|
|
img = img.filter(ImageFilter.GaussianBlur(radius=blur))
|
||
|
|
|
||
|
|
return np.asarray(img)/256.
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def load_image_as_linear_sRGB(path, blur = 0.0):
|
||
|
|
|
||
|
|
path = Path(path)
|
||
|
|
location = str(path.parents[0])
|
||
|
|
name = str(path.name)#be sure that image doesn't lose quality
|
||
|
|
|
||
|
|
print("proccesing " + name)
|
||
|
|
img = Image.open(path)
|
||
|
|
|
||
|
|
if blur != 0.0:
|
||
|
|
img = img.filter(ImageFilter.GaussianBlur(radius=blur))
|
||
|
|
|
||
|
|
img_array = np.asarray(img)/256.
|
||
|
|
img_sRGB_linear_array = sRGB_to_sRGB_linear(img_array)
|
||
|
|
return img_sRGB_linear_array
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|