From a0111439b27634f88a2b3fc69c220400192b8b75 Mon Sep 17 00:00:00 2001 From: ruri Date: Wed, 29 May 2019 15:19:52 +0800 Subject: [PATCH] fix infer, add softmax (#2330) --- PaddleCV/image_classification/infer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/PaddleCV/image_classification/infer.py b/PaddleCV/image_classification/infer.py index e6e126f2..0ad94ff3 100644 --- a/PaddleCV/image_classification/infer.py +++ b/PaddleCV/image_classification/infer.py @@ -47,6 +47,7 @@ def infer(args): out, _, _ = model.net(input=image, class_dim=class_dim) else: out = model.net(input=image, class_dim=class_dim) + out = fluid.layers.softmax(out) test_program = fluid.default_main_program().clone(for_test=True) -- GitLab