import numpy as np
from PIL import Image

# originál
img = Image.open('../kvetina.jpg')
data = np.asarray(img)
print(data.shape)   # (375, 500, 3)

# solarizace
# ~ pro pixely, jejichž hodnota je větší než 77, proveď barevnou negaci
# ~ ostatní pixely zachovej v původní barvě
# a) > 77
solarized_data = np.where(data > 77, 255 - data, data)
Image.fromarray(solarized_data, 'RGB').save('solarized_1.jpg')
# b) <= 77
solarized_data = np.where(data <= 77, 255 - data, data)
Image.fromarray(solarized_data, 'RGB').save('solarized_2.jpg')