summaryrefslogtreecommitdiff
path: root/openCVTools.py
blob: 484c3fa4412b7616a159a8f95d8a282b37be94ee (plain)
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
38
39
40
41
42
43
44
45
import cv2 as cv
import sys
import numpy as np
import matplotlib.pyplot as plt


def safeLoad(pathToFile):
    '''
    OpenCV does no validation checks due to performance reasons.
    Therefore, this function checks if the image could be loaded
    '''
    img = cv.imread(pathToFile)
    if img is None:
        sys.exit("Image could not be loaded.")
    return img


# TODO Aufgabe 1
'''
Passen Sie die Funktion `imageStats(..)` so an, dass sie sowohl Grau- als auch Farbbilder korrekt anzeigt.
Erweitern Sie die Funktion zusätzlich so dass der Datentyp mit ausgegeben wird.
'''
def imageStats(img):
    '''
    Returns a few image statistics
    '''
    s = img.shape
    return f'Width: {s[1]}, Height: {s[0]}, Channels: {s[2]}'



# TODO Aufgabe 1
'''
Passen Sie die Funktion `showImage(..)` so an, dass sie sowohl Grau- als auch Farbbilder korrekt anzeigt.
'''
def showImage(title, originalImg):
    print(imageStats(originalImg))

    img = originalImg.copy()
    img = img[:,:,::-1]
    plt.figure(title)
    plt.imshow(img)
    plt.show()