handle case when there are no labels

2026-07-07 08:48:01 +08:00 · 2013-10-07 23:17:52 -07:00
parent 794176b7c4
commit 4db1e1b83c
1 changed files with 3 additions and 0 deletions
@@ -101,6 +101,9 @@ def label2rgb(label, image=None, colors=None, alpha=0.3,
            bg_color = _rgb_vector(bg_color)
            color_cycle = itertools.chain(bg_color, color_cycle)

+    if len(labels) == 0:
+        return img_layer
+
    label_to_color = np.zeros((max(labels) + 1, 3))
    for lab, c in zip(labels, color_cycle):
        label_to_color[lab] = c