From 43ce1a629040be2b61cfc20c1b1ba333b64d2721 Mon Sep 17 00:00:00 2001 From: MRXLT Date: Mon, 1 Jun 2020 11:59:17 +0000 Subject: [PATCH] fix resnet v2 demo --- python/examples/resnet_v2_50/resnet50_v2_tutorial.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/examples/resnet_v2_50/resnet50_v2_tutorial.py b/python/examples/resnet_v2_50/resnet50_v2_tutorial.py index 8d916cbd..b249d2a6 100644 --- a/python/examples/resnet_v2_50/resnet50_v2_tutorial.py +++ b/python/examples/resnet_v2_50/resnet50_v2_tutorial.py @@ -14,7 +14,7 @@ from paddle_serving_client import Client from paddle_serving_app.reader import Sequential, File2Image, Resize, CenterCrop -from apddle_serving_app.reader import RGB2BGR, Transpose, Div, Normalize +from paddle_serving_app.reader import RGB2BGR, Transpose, Div, Normalize client = Client() client.load_client_config( @@ -28,5 +28,5 @@ seq = Sequential([ image_file = "daisy.jpg" img = seq(image_file) -fetch_map = client.predict(feed={"image": img}, fetch=["feature_map"]) -print(fetch_map["feature_map"].reshape(-1)) +fetch_map = client.predict(feed={"image": img}, fetch=["score"]) +print(fetch_map["score"].reshape(-1)) -- GitLab