我正在处理一个实例分割任务,我已经为我的面具生成了注释。我目前有一个 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的标签功能。然而,这里的问题是它为每个非连接的注释分配了不同的值(即第一张图像中的注释与第二张图像中的注释不同,但第二张图像中的注释都是单个(非连接的) ) 注释。标签函数将它们视为不同的对象而不是相同的对象)。
任何有关如何解决或解决此问题的帮助将不胜感激!!谢谢!