fix mnist predict.py bug

image_classification/index.html

@@ -229,7 +229,9 @@ def process(settings, file_list):
             labels = batch['labels']
             for im, lab in zip(images, labels):
                 if settings.is_train and np.random.randint(2):
+                    im = im.reshape(3, 32, 32)
                     im = im[:,:,::-1]
+                    im = im.flatten()
                 im = im - settings.mean
                 yield {
                     'image': im.astype('float32'),

recognize_digits/README.md

@@ -389,7 +389,7 @@ The classification accuracy is 99.20%
 脚本  `predict.py` 可以对训练好的模型进行预测，例如softmax回归中：
 
 ```bash
-python predict.py -c softmax_mnist.py -d data/raw_data/ -m softmax_mnist_model/pass-00047
+python predict.py -c mnist_model.py -d data/raw_data/ -m softmax_mnist_model/pass-00047
 ```
 
 - -c 指定模型的结构

recognize_digits/index.html

@@ -430,7 +430,7 @@ The classification accuracy is 99.20%
 脚本  `predict.py` 可以对训练好的模型进行预测，例如softmax回归中：
 
 ```bash
-python predict.py -c softmax_mnist.py -d data/raw_data/ -m softmax_mnist_model/pass-00047
+python predict.py -c mnist_model.py -d data/raw_data/ -m softmax_mnist_model/pass-00047
 ```
 
 - -c 指定模型的结构

recognize_digits/predict.py

@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-"""Usage: predict.py -c CONF -d ./data/raw_data/  -m MODEL
+"""Usage: predict.py -c CONF -d DATA  -m MODEL
 
 
 Arguments: