forked from douxbb/refactoring17
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathfilter.py
More file actions
35 lines (35 loc) · 1.89 KB
/
filter.py
File metadata and controls
35 lines (35 loc) · 1.89 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import numpy as np
from PIL import Image
class GreyImage:
def __init__(self, pix_array, pix_size=10, gradation=5):
self.image = np.array(pix_array)
self.size = pix_size
self.grad = 255 // gradation
self.width = len(self.image)
self.height = len(self.image[0])
def get_grey_image(self):
for x in range(0, self.width, self.size):
for y in range(0, self.height, self.size):
self.image[x:x + self.size, y:y + self.size] = self.get_middle_color(x, y)
return Image.fromarray(self.image)
def get_middle_color(self, x, y):
return int(self.image[x:x + self.size, y:y + self.size].sum() / 3 // self.size ** 2 // self.grad * self.grad)
flag = 'no'
while flag == 'no':
print('Введите название картинки')
orig_img = Image.open('{}'.format(input()))
print('Выберите величину ячейки, на которую будут делить изображение\n' +
'Размер должен быть инициализирован целым положительным числом')
pic_size = int(input())
print('Выберите количество оттенков\n' +
'Количество должно быть инициализировано целым положительным числом')
grad = int(input())
arr = GreyImage(orig_img, pix_size=pic_size, gradation=grad).get_grey_image()
print('Введите название новой картинки')
name = input()
print('Введите формат новой картинки(jpg, png и т.д.)')
formt = input()
arr.save('{}.{}'.format(name, formt))
print('Если хотите завершить сессию, наберите \'yes\', если нет-нажмите enter')
if input() == 'yes':
flag = 'yes'