Add comment to apply_mask_on_image function
parent
bfa7eb87ad
commit
1bdf70da5b
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue