From b781b4150cf86b78226b616bb68cbad99f133ad2 Mon Sep 17 00:00:00 2001 From: Leonard Kugis Date: Tue, 22 Nov 2022 08:37:34 +0100 Subject: Initial commit --- openCVTools.py | 45 +++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 45 insertions(+) create mode 100644 openCVTools.py (limited to 'openCVTools.py') diff --git a/openCVTools.py b/openCVTools.py new file mode 100644 index 0000000..484c3fa --- /dev/null +++ b/openCVTools.py @@ -0,0 +1,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() + + -- cgit v1.2.1