modify inputs to image

2021-11-26 18:34:39 +00:00 · 2021-11-26 18:34:39 +00:00 · 654a9deebf
parent b8b7ae4012
commit 654a9deebf
1 changed files with 1 additions and 1 deletions
--- a/deploy/paddleserving/classification_web_service.py
+++ b/deploy/paddleserving/classification_web_service.py
@ -47,7 +47,7 @@ class ImagenetOp(Op):
            img = self.seq(im)
            imgs.append(img[np.newaxis, :].copy())
        input_imgs = np.concatenate(imgs, axis=0)
-        return {"inputs": input_imgs}, False, None, ""
+        return {"image": input_imgs}, False, None, ""

    def postprocess(self, input_dicts, fetch_dict, data_id, log_id):
        score_list = fetch_dict["prediction"]