fix representation problems

cc901a63 · JiabinYang · ee4e2188 · cc901a63
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

03.image_classification/train.py 03.image_classification/train.py +1 -0

未找到文件。
--- a/03.image_classification/train.py
+++ b/03.image_classification/train.py
@@ -116,6 +116,7 @@ def infer(use_cuda, inference_program, params_dirname=None):
    img = load_image(cur_dir + '/image/dog.png')

    # inference
+    results = inferencer.infer({'pixel': img})

    label_list = [
        "airplane", "automobile", "bird", "cat", "deer", "dog", "frog", "horse",