Readd reshape to base preprocessor

Seems to save a lot of storage space for some unknown reason
jacobbieker · Oct 26, 2018 · e8e8745 · e8e8745
1 parent e20ae3f
commit e8e8745
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/factnn/preprocess/base_preprocessor.py b/factnn/preprocess/base_preprocessor.py
@@ -148,6 +148,7 @@ def reformat(self, image):
         :return:
         """
         dataset = np.swapaxes(image, 1, 3)
+        dataset = np.array(dataset).reshape((self.shape[0], self.shape[3], self.shape[2], self.shape[1])).astype(np.float32)
         return dataset
 
     def format(self, batch):