From 1bdf70da5bf67267844bb7be919115ffea0aafba Mon Sep 17 00:00:00 2001 From: Gasper Spagnolo Date: Sat, 22 Oct 2022 18:10:50 +0200 Subject: [PATCH] Add comment to apply_mask_on_image function --- assignment1/uz_framework/image.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/assignment1/uz_framework/image.py b/assignment1/uz_framework/image.py index a0db909..ea4d842 100644 --- a/assignment1/uz_framework/image.py +++ b/assignment1/uz_framework/image.py @@ -213,6 +213,9 @@ def get_image_bins(image: Union[npt.NDArray[np.float64], npt.NDArray[np.uint8]] return counts / np.sum(counts) def apply_mask_on_image(image: Union[npt.NDArray[np.float64], npt.NDArray[np.uint8]], mask: npt.NDArray[np.uint8]): + """ + Accepts image and applys mask to image + """ image = image.copy() mask = np.expand_dims(mask, axis=2)