Merge pull request #594 from JiabinYang/book03_refine

fix representation problem in train.py

03.image_classification/train.py

@@ -118,7 +118,11 @@ def infer(use_cuda, inference_program, params_dirname=None):
     # inference
     results = inferencer.infer({'pixel': img})
 
-    print("infer results: ", results)
+    label_list = [
+        "airplane", "automobile", "bird", "cat", "deer", "dog", "frog", "horse",
+        "ship", "truck"
+    ]
+    print("infer results: %s" % label_list[np.argmax(results[0])])
 
 
 def main(use_cuda):