From 440f2ba69efaa58d45587f53c2729e07c96859ae Mon Sep 17 00:00:00 2001 From: Jiawei Wang Date: Wed, 9 Dec 2020 14:33:05 +0800 Subject: [PATCH] Update test_client.py --- python/examples/cascade_rcnn/test_client.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/examples/cascade_rcnn/test_client.py b/python/examples/cascade_rcnn/test_client.py index 75e25985..b40e97ac 100644 --- a/python/examples/cascade_rcnn/test_client.py +++ b/python/examples/cascade_rcnn/test_client.py @@ -33,7 +33,7 @@ fetch_map = client.predict( "im_shape": np.array(list(im.shape[1:]) + [1.0]) }, fetch=["multiclass_nms_0.tmp_0"], - batch=True) + batch=False) fetch_map["image"] = '000000570688.jpg' print(fetch_map) postprocess(fetch_map) -- GitLab