bug fix with mean correction.

ragavvenkatesan · Jan 24, 2017 · 9f224d1 · 9f224d1
1 parent d124178
commit 9f224d1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/yann/utils/dataset.py b/yann/utils/dataset.py
@@ -733,11 +733,11 @@ def preprocessing( data, height, width, channels, args):
 	data = numpy.reshape(data,out_shp)
 	if mean_subtract is True:
 		if normalize is True or ZCA is True: 
-			data = (data - data.mean(axis = 0)) / (data.max(axis = 0) + 1e-7)
+			data = (data - data.mean() / (data.max() + 1e-7)
 			# do this normalization thing in batch mode.	
 	else:
 		if normalize is True or ZCA is True:
-			data = data / (data.max(axis =0) + 1e-7)
+			data = data / (data.max() + 1e-7)
 
 	if ZCA is True: