Merge pull request NVIDIA#1290 from gheinrich/dev/segmentation-extens…

…ion-helpers Add info to image segmentation data extension
ethantang95 · Nov 22, 2016 · 1bce875 · 1bce875
2 parents c5fb49c + 7b13911
commit 1bce875
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/digits/extensions/data/imageSegmentation/data.py b/digits/extensions/data/imageSegmentation/data.py
@@ -55,7 +55,12 @@ def __init__(self, **kwargs):
                 lines = f.read().splitlines()
                 for line in lines:
                     for val in line.split():
-                        palette.append(int(val))
+                        try:
+                            palette.append(int(val))
+                        except:
+                            raise ValueError("Your color map file seems to "
+                                             "be badly formatted: '%s' is not "
+                                             "an integer" % val)
                 # fill rest with zeros
                 palette = palette + [0] * (256 * 3 - len(palette))
                 self.userdata[COLOR_PALETTE_ATTRIBUTE] = palette
@@ -189,11 +194,15 @@ def load_label(self, filename):
         if self.userdata['colormap_method'] == "label":
             if image.mode not in ['P', 'L', '1']:
                 raise ValueError("Labels are expected to be single-channel (paletted or "
-                                 " grayscale) images - %s mode is '%s'"
+                                 " grayscale) images - %s mode is '%s'. If your labels are "
+                                 "RGB images then set the 'Color Map Specification' field "
+                                 "to 'from text file' and provide a color map text file."
                                  % (filename, image.mode))
         else:
             if image.mode not in ['RGB']:
-                raise ValueError("Labels are expected to be RGB images - %s mode is '%s'"
+                raise ValueError("Labels are expected to be RGB images - %s mode is '%s'. "
+                                 "If your labels are palette or grayscale images then set "
+                                 "the 'Color Map Specification' field to 'from label image'."
                                  % (filename, image.mode))
             image = image.quantize(palette=self.palette_img)
 

diff --git a/digits/extensions/data/imageSegmentation/template.html b/digits/extensions/data/imageSegmentation/template.html
@@ -94,3 +94,7 @@
 $("#colormap_method").click(hasColorMapMethodChanged);
 hasColorMapMethodChanged();
 </script>
+
+<div>
+    Note: the recommended label encoding is PNG.
+</div>