uz_assignments/assignment1/solution.py

import UZ_utils as uz
import numpy as np
import numpy.typing as npt

#######################################
# EXCERCISE 1: Basic image processing #
#######################################

def excercise_one() -> None:
    image = one_a()
    one_b(image)
    
# Display image
def one_a() -> npt.NDArray[np.float64]:
    image = uz.imread('./images/bird.jpg')
    uz.imshow(image, 'Birdie')
    return image

# Convert loaded image into grayscale
def one_b(image: npt.NDArray[np.float64]) -> None:
    grayscale_image = np.zeros(image.shape[:2])
    
    for i in range(image.shape[0]):
        for j in range(image.shape[1]):
            grayscale_image[i, j] = (image[i, j, 0] + image[i,j, 1] + image[i, j, 2]) / 3

    uz.imshow(grayscale_image, 'Birdie grayscale')

def one_c():
    #todo
    assert(1 == 1)
    
def main() -> None:
    excercise_one() 

if __name__ == "__main__":
    main()