image-processing-basics/utils.py at main · lawtherea/image-processing-basics · GitHub

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
from PIL import Image
import numpy as np
import os

input_folder = "inputs"
output_folder = "outputs"

def in_path(filename):
    return os.path.join(input_folder, filename)

def out_path(filename):
    return os.path.join(output_folder, filename)

# compare two images side by side
def show_vertical(img1, img2):
    im = Image.fromarray(np.vstack((np.array(img1), np.array(img2))))
    im.show()
    return im

def show_horizontal(img1, img2):
    im = Image.fromarray(np.hstack((np.array(img1), np.array(img2))))
    im.show()
    return im

# compare 3 images side by side
def show_horizontal_three(img1, img2, img3):
    im = Image.fromarray(np.hstack((np.array(img1), np.array(img2), np.array(img3))))
    im.show()
    return im

def show_horizontal_four(img1, img2, img3, img4):
    im = Image.fromarray(np.hstack((np.array(img1), np.array(img2), np.array(img3), np.array(img4))))
    im.show()
    return im