Merge pull request #73 from dayhaha/fix_mnist_bug

remove matplotlib from predict.py

recognize_digits/predict.py

@@ -13,7 +13,6 @@
 # limitations under the License.
 """Usage: predict.py -c CONF -d DATA -m MODEL
 
-
 Arguments:
     CONF        train conf
     DATA        MNIST Data
@@ -29,7 +28,6 @@ Options:
 import os
 import sys
 from docopt import docopt
-import matplotlib.pyplot as plt
 import numpy as np
 
 from py_paddle import swig_paddle, DataProviderConverter
@@ -48,6 +46,7 @@ class Prediction():
         self.network.loadParameters(model_dir)
 
         self.images, self.labels = read_data(data_dir, "t10k")
+        self.images = self.images / 255.0 * 2.0 - 1.0  # normalized to [-1,1]
 
         slots = [dense_vector(28 * 28)]
         self.converter = DataProviderConverter(slots)