Update classification_web_service.py

2021-12-16 14:21:01 +08:00 · 2021-12-16 14:21:01 +08:00 · a64c12385d
parent d4954e1df5
commit a64c12385d
1 changed files with 1 additions and 1 deletions
--- a/deploy/paddleserving/classification_web_service.py
+++ b/deploy/paddleserving/classification_web_service.py
@ -49,7 +49,7 @@ class ImagenetOp(Op):
        input_imgs = np.concatenate(imgs, axis=0)
        return {"inputs": input_imgs}, False, None, ""

-    def postprocess(self, input_dicts, fetch_dict, log_id):
+    def postprocess(self, input_dicts, fetch_dict, data_id, log_id):
        score_list = fetch_dict["prediction"]
        result = {"label": [], "prob": []}
        for score in score_list: