FAKS
/
uz_assignments
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add comment to apply_mask_on_image function

main
Gasper Spagnolo 2022-10-22 18:10:50 +02:00
parent bfa7eb87ad
commit 1bdf70da5b
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
assignment1/uz_framework/image.py
View File

@ -213,6 +213,9 @@ def get_image_bins(image: Union[npt.NDArray[np.float64], npt.NDArray[np.uint8]]
return counts / np.sum(counts) 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]): 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() image = image.copy()
mask = np.expand_dims(mask, axis=2) mask = np.expand_dims(mask, axis=2)