-
Notifications
You must be signed in to change notification settings - Fork 1
/
blockify.py
37 lines (26 loc) · 1.17 KB
/
blockify.py
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
36
37
import cv2
import numpy as np
from PIL import Image, ImageDraw
def convert(img_loc):
image = cv2.imread(img_loc)
image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
height, width, garbage = image.shape
length = 7
num_cols = int(width / length)
num_rows = int(height / length)
out_width = length * num_cols
out_height = length * num_rows
out_img = Image.new("RGB", (out_width, out_height))
draw = ImageDraw.Draw(out_img)
for i in range(int(num_rows)):
for j in range(int(num_cols)):
partial_image = image[int(i * length): min(int((i + 1) * length), height),
int(j * length): min(int((j + 1) * length), width),
:]
partial_avg_color = np.sum(np.sum(partial_image, axis=0), axis=0) / (length * length)
partial_avg_color = tuple(partial_avg_color.astype(np.int32).tolist())
draw.rectangle((int(j * length), int(i * length), min(int((j + 1) * length), width),
min(int((i + 1) * length), height)), fill=partial_avg_color)
cropped_img = out_img.getbbox()
f_img = out_img.crop(cropped_img)
return f_img