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

依賴
- 請先安裝 capybara-docsaid[visualization]。
說明：在影像上繪製遮罩。
參數
- img (np.ndarray)：要繪製的影像。
- mask (np.ndarray)：要繪製的遮罩。
- colormap (int)：用於遮罩的色彩地圖。預設為 cv2.COLORMAP_JET。
- weight (Tuple[float, float])：影像和遮罩的權重。預設為 (0.5, 0.5)。
- gamma (float)：遮罩的 Gamma 值。預設為 0。
- min_max_normalize (bool)：是否將遮罩正規化為範圍 [0, 1]。預設為 False。
傳回值
- np.ndarray：繪製了遮罩的影像。

範例

import cv2
import numpy as np
from capybara import Polygon, imread
from capybara.vision.visualization.draw import draw_mask, draw_polygon

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

draw_mask