draw_mask

draw_mask(img: np.ndarray, mask: np.ndarray, colormap: int = cv2.COLORMAP_JET, weight: Tuple[float, float] = (0.5, 0.5), gamma: float = 0, min_max_normalize: bool = False) -> np.ndarray

• Description: Draws a mask on an image.

• Parameters:

  • img (np.ndarray): The image to draw on.
  • mask (np.ndarray): The mask to draw.
  • colormap (int): The colormap to apply to the mask. Defaults to cv2.COLORMAP_JET.
  • weight (Tuple[float, float]): The weights for the image and the mask. Defaults to (0.5, 0.5).
  • gamma (float): The gamma value for the mask. Defaults to 0.
  • min_max_normalize (bool): Whether to normalize the mask to the range [0, 1]. Defaults to False.

• Return Value:

  • np.ndarray: The image with the mask drawn on it.

• Example:

  import capybara as cb
  import numpy as np
  
  img = cb.imread('lena.png')
  polygon = cb.Polygon([(20, 20), (100, 20), (80, 80), (20, 40)])
  mask = cb.draw_polygon(np.zeros_like(img), polygon, fillup=True, color=255)
  mask_img = cb.draw_mask(img, mask, colormap=cv2.COLORMAP_JET, weight=(0.5, 0.5), gamma=0, min_max_normalize=False)