From edc9a234a1b46e8ff870b7ce740458b6c98deaff Mon Sep 17 00:00:00 2001 From: dongdaxiang Date: Mon, 11 May 2020 16:05:24 +0800 Subject: [PATCH] fix code style --- .../examples/faster_rcnn_model/new_test_client.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/python/examples/faster_rcnn_model/new_test_client.py b/python/examples/faster_rcnn_model/new_test_client.py index ff4ebf37..0c6c615f 100755 --- a/python/examples/faster_rcnn_model/new_test_client.py +++ b/python/examples/faster_rcnn_model/new_test_client.py @@ -23,11 +23,18 @@ preprocess = Sequential([ postprocess = RCNNPostprocess("label_list.txt", "output") client = Client() -client.load_client_config("faster_rcnn_client_conf/serving_client_conf.prototxt") + +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 = 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[1] postprocess(fetch_map) -- GitLab