Update utils.py

JNingWei · Dec 22, 2017 · 9f93990 · 9f93990
1 parent 5d19bca
commit 9f93990
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/FCN/src/utils.py b/FCN/src/utils.py
@@ -1,9 +1,16 @@
+# coding=utf-8
+
+"""
+Image-Segmentation
+Expansion package.
+__author__ = 'JNingWei'
+"""
+
 import numpy as np
 import matplotlib as mpl
 import matplotlib.cm
 
 def color_image(image, num_classes=20):
-
     norm = mpl.colors.Normalize(vmin=0., vmax=num_classes)
     mycm = mpl.cm.get_cmap('Set1')
     return mycm(norm(image))