From 0bc9660fccdaaf45947d0f97c59bbeb10283f176 Mon Sep 17 00:00:00 2001 From: "lee.lcy" Date: Thu, 20 Oct 2022 21:12:14 +0800 Subject: [PATCH] [to #42322933] modify img_embedding type for image_reid_person modify img_embedding type for image_reid_person Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10477972 --- modelscope/pipelines/cv/image_reid_person_pipeline.py | 1 + 1 file changed, 1 insertion(+) diff --git a/modelscope/pipelines/cv/image_reid_person_pipeline.py b/modelscope/pipelines/cv/image_reid_person_pipeline.py index 64674a65..9f60142a 100644 --- a/modelscope/pipelines/cv/image_reid_person_pipeline.py +++ b/modelscope/pipelines/cv/image_reid_person_pipeline.py @@ -53,6 +53,7 @@ class ImageReidPersonPipeline(Pipeline): def forward(self, input: Dict[str, Any]) -> Dict[str, Any]: img = input['img'] img_embedding = self.model(img) + img_embedding = img_embedding.detach().cpu().numpy() return {OutputKeys.IMG_EMBEDDING: img_embedding} def postprocess(self, inputs: Dict[str, Any]) -> Dict[str, Any]: