init

app.py

@@ -23,8 +23,8 @@ from skimage import io, segmentation, morphology, measure, exposure
 def visualize_instance_seg_mask(mask):
     image = np.zeros((mask.shape[0], mask.shape[1], 3))
     labels = np.unique(mask)
-    label2color = {0: (0, 0, 0)}
     label2color = {label: (random.randint(0, 1), random.randint(0, 255), random.randint(0, 255)) for label in labels if label > 0}
+    label2color[0] = (0, 0, 0)
     for i in range(image.shape[0]):
       for j in range(image.shape[1]):
         image[i, j, :] = label2color[mask[i, j]]