Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-24 20:19:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

[to #42322933] modify img_embedding type for image_reid_person

Browse Source 
modify img_embedding type for image_reid_person
        Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10477972
This commit is contained in:
lee.lcy
2022-10-20 21:12:14 +08:00
committed by yingda.chen
parent 2b49b322a2
commit 0bc9660fcc
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modelscope/pipelines/cv/image_reid_person_pipeline.py
View File

@@ -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]: