Added comments

madhavkhoslaa · Jul 23, 2019 · 9b9f88d · 9b9f88d
1 parent 0522cc6
commit 9b9f88d
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 13 deletions.
diff --git a/.DS_Store b/.DS_Store
diff --git a/test.py b/test.py
diff --git a/utils/one_hot_encoder.py b/utils/one_hot_encoder.py
@@ -51,6 +51,8 @@ def PerPixelClassMatrix(self, Image):
                 class_map[x][y]= self.color[clr]
         return class_map
     def HotEncode(self, class_map):
+        """Inputs the class map of the image and returns the hot encoded
+        N channel output. Each channel represents the presence of a class per pixel"""
         assert isinstance(class_map, np.ndarray), "Class map has to be an ndarray and not {}".format(type(class_map))
         shape_= class_map.shape
         encoded= np.zeros(shape= (shape_[0], shape_[1], self.n_classes), dtype=float)