diff --git a/PaddleCV/image_classification/infer.py b/PaddleCV/image_classification/infer.py
index e6e126f259c15dfe62acba167a04fe5caed58ad7..0ad94ff320f83bb8f733976b01af3204474fb3ec 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)