From 7cc40ddbff479cc477f37bb37cd368b577af88aa Mon Sep 17 00:00:00 2001 From: Juan Nunez-Iglesias Date: Wed, 18 Jun 2014 16:44:02 +1000 Subject: [PATCH] Add background label handling for label2rgb_avg --- skimage/color/colorlabel.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/skimage/color/colorlabel.py b/skimage/color/colorlabel.py index 47cb0de9..8074c2a3 100644 --- a/skimage/color/colorlabel.py +++ b/skimage/color/colorlabel.py @@ -159,8 +159,10 @@ def _label2rgb_overlay(label, image=None, colors=None, alpha=0.3, def _label2rgb_avg(label_field, image, bg_label, bg_color): out = np.zeros_like(image) labels = np.unique(label_field) - if (labels == bg_label).any(): + bg = (labels == bg_label) + if bg.any(): labels = labels[labels != bg_label] + out[bg] = bg_color for label in labels: mask = (label_field == label).nonzero() color = image[mask].mean(axis=0)