fix representation problems

JiabinYang · Aug 16, 2018 · cc901a6 · cc901a6
1 parent ee4e218
commit cc901a6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/03.image_classification/train.py b/03.image_classification/train.py
@@ -116,6 +116,7 @@ def load_image(file):
     img = load_image(cur_dir + '/image/dog.png')
 
     # inference
+    results = inferencer.infer({'pixel': img})
 
     label_list = [
         "airplane", "automobile", "bird", "cat", "deer", "dog", "frog", "horse",