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)
|
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)
|
||||||
|
|
Loading…
Reference in New Issue