如何用不同的颜色给掩码注释上色(填充)

问题描述 投票:0回答:0

我正在处理一个实例分割任务,我已经为我的面具生成了注释。我目前有一个 for 循环遍历所有注释,并将它们绘制在空白(即 numpy.zeros((image.shape)))画布上,从而在第一个循环中,掩码(带有具有值的注释 = (1, 1, 1)) 看起来像这样

在第二个循环中,掩码(带有新注释但值相同)看起来像这样。

我正在尝试用不同的颜色为每个注释着色(或“填写”不确定正确的术语是什么)(如下面的示例图像)。我最初尝试这样做

import numpy as np
import skimage

mask_array = np.zeros((800, 1200, 3))
for annotation in annotations:
    # draw the annotation on the mask
    mask_array = annotation.value.draw(canvas=mask_array, thickness=5)
    
    # trying to fill the annotation with a different colour
    mask_array = skimage.morphology.label(mask_array)

那是我尝试使用skimage的标签功能。然而,这里的问题是它为每个非连接的注释分配了不同的值(即第一张图像中的注释与第二张图像中的注释不同,但第二张图像中的注释都是单个(非连接的) ) 注释。标签函数将它们视为不同的对象而不是相同的对象)。

任何有关如何解决或解决此问题的帮助将不胜感激!!谢谢!

我举了一个我想要实现的例子。

python image numpy deep-learning computer-vision
© www.soinside.com 2019 - 2024. All rights reserved.