indent

madhavkhoslaa · Jul 22, 2019 · ce4bc75 · ce4bc75
1 parent 8e09752
commit ce4bc75
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/utils/one_hot_encoder.py b/utils/one_hot_encoder.py
@@ -60,7 +60,7 @@ def HotEncode(self, class_map):
                 encoded[x][y][category]= 1
         return encoded
 
-        def channelSplit(self,Nch_Hot_encoded):
-            """The hot encoded Image is stored as stack of Hot encoded ndarrays
-                this function returns a list of cll the channels"""
-            return np.dsplit(Nch_Hot_encoded,Nch_Hot_encoded.shape[-1])
+    def channelSplit(self,Nch_Hot_encoded):
+        """The hot encoded Image is stored as stack of Hot encoded ndarrays
+            this function returns a list of cll the channels"""
+        return np.dsplit(Nch_Hot_encoded,Nch_Hot_encoded.shape[-1])