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)
    one_c(image)
    one_d(60, 180, 180, 340, 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')

# Display birdies head in grayscale
def one_c(image: npt.NDArray[np.float64]) -> None:
    print(image.shape)
    #uz.imshow(image[50:200, 100:400, 2])

# Convert one part of image to grayscale
def one_d(startx: int, endx: int, starty: int, endy: int, image:npt.NDArray[np.float64]) -> None:

    for i in range(startx, endx):
        for j in range(starty, endy):
            image[i, j, 0] = 1 - image[i, j, 0]
            image[i, j, 1] = 1 - image[i, j, 1]
            image[i, j, 2] = 1 - image[i, j, 2]

    uz.imshow(image, 'Bridie head grayscale')

    
def main() -> None:
    excercise_one() 

if __name__ == "__main__":
    main()