Make utils compatible with newest matplotlib Version.

matplotlib.cm is now imported explicitly.
MarvinTeichmann · Jan 11, 2017 · abd5ff9 · abd5ff9
1 parent 489d811
commit abd5ff9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/utils.py b/utils.py
@@ -3,6 +3,7 @@
 
 def color_image(image, num_classes=20):
     import matplotlib as mpl
+    import matplotlib.cm
     norm = mpl.colors.Normalize(vmin=0., vmax=num_classes)
     mycm = mpl.cm.get_cmap('Set1')
     return mycm(norm(image))