使用Python与OpenCV实现多实例掩膜图像的绘制

def create_color_palette(num_colors, seed=47):
    """
    生成随机颜色列表
    """
    np.random.seed(seed)
    palette = []
    for _ in range(num_colors):
        color = (np.random.rand(3,) * 0.8).astype(int)
        palette.append(color.tolist())

    return palette


def render_instance_masks(image, instance_masks, border_thickness=1):
    """
    在给定图像上渲染多个实例的掩膜及其边界。
    :param image: 原始三通道图像
    :param instance_masks: 包含多个实例的掩膜数据
    :param border_thickness: 边界线宽度
    :return: 添加了实例掩膜及边界的图像
    """
    output_image = image.copy()

    # 分割掩膜
    height, width = instance_masks.shape[:2]
    unique_ids = np.unique(instance_masks)
    if unique_ids[0] == 0:
        unique_ids = unique_ids[1:]  # 排除背景
    masks = [instance_masks == uid for uid in unique_ids]

    # 应用颜色到每个实例
    colors = create_color_palette(len(unique_ids))
    for mask, color in zip(masks, colors):
        output_image[mask] = color

    # 图像融合
    blended_image = cv2.addWeighted(image, 0.5, output_image, 0.5, 0)

    # 绘制边界线
    for mask in masks:
        contour_mask = np.zeros((height, width), dtype=np.uint8)
        contour_mask[mask] = 255
        contours, _ = cv2.findContours(contour_mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
        cv2.drawContours(blended_image, contours, -1, (255, 255, 255), border_thickness)

    return blended_image