From 4e971502b246b023351f4a292f0bda3843984104 Mon Sep 17 00:00:00 2001 From: Dong Daxiang <35550832+guru4elephant@users.noreply.github.com> Date: Sat, 9 May 2020 17:28:39 +0800 Subject: [PATCH] Update new_test_client.py --- python/examples/faster_rcnn_model/new_test_client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/examples/faster_rcnn_model/new_test_client.py b/python/examples/faster_rcnn_model/new_test_client.py index eeded665..197530a6 100755 --- a/python/examples/faster_rcnn_model/new_test_client.py +++ b/python/examples/faster_rcnn_model/new_test_client.py @@ -24,11 +24,11 @@ preprocess = Sequential([ postprocess = RCNNPostprocess("label_list.txt", "output") client = Client() -client.load_client_config(sys.argv[1]) +client.load_client_config("faster_rcnn_client_conf/serving_client_conf.prototxt") client.connect(['127.0.0.1:9393']) im = preprocess(sys.argv[2]) fetch_map = client.predict(feed={"image": im, "im_info": np.array(list(im.shape[1:]) + [1.0]), "im_shape": np.array(list(im.shape[1:]) + [1.0])}, fetch=["multiclass_nms"]) -fetch_map["image"] = sys.argv[2] +fetch_map["image"] = sys.argv[1] postprocess(fetch_map) -- GitLab