filter.py

import doctest

from PIL import Image
import numpy as np

def get_illumination(pixels, p_x, p_y, size):
    """
    Считает компонет серого цвета и возвращает среднюю яркость
    >>> get_illumination(np.array(Image.open('img2.jpg')), 7, 4, 9)
    18
    >>> get_illumination(np.array(Image.open('img2.jpg')), 15, 10, 10)
    19
    """
    summa = np.sum(pixels[p_x: p_x + size, p_y: p_y + size])
    result = int(summa // 3)
    return int(result // (size * size))

def install_color(pixels, illumination, size, p_x, p_y, step):
    """
    Закрашивает пиксели в один цвет
    """
    pixels[p_x: p_x + size, p_y: p_y + size] = int(illumination // step) * step

def get_gray(pixels, gradation, size):
    """
    Возвращает итоговое изображение
    """
    step = 255 // (gradation - 1)
    hei = len(pixels)
    wid = len(pixels[1])
    for y in range(0, hei, size):
        for x in range(0, wid, size):
            illumination = get_illumination(pixels, x, y, size)
            install_color(pixels, illumination, size, x, y, step)
    return pixels

if __name__ == '__main__':
    doctest.testmod()

img = Image.open('img2.jpg')
pixels = np.array(img)
res = Image.fromarray(get_gray(pixels,
                                int(input("Введите размер градации")), 
                                int(input("Введите размер мозаики"))))
res.save('res1.jpg')