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

[to #42322933] fix cv/animalRecog output format

Browse Source 
* fix output format
        Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10491722
This commit is contained in:
yanheng.wyh
2022-10-23 12:22:33 +08:00
committed by yingda.chen
parent 182ba1768f
commit cf831dbf98
2 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
modelscope/pipelines/cv/animal_recognition_pipeline.py
View File

@@ -113,9 +113,8 @@ class AnimalRecognitionPipeline(Pipeline):
label_mapping = f.readlines()
score = torch.max(inputs['outputs'])
inputs = {
OutputKeys.SCORES:
score.item(),
OutputKeys.SCORES: [score.item()],
OutputKeys.LABELS:
label_mapping[inputs['outputs'].argmax()].split('\t')[1]
[label_mapping[inputs['outputs'].argmax()].split('\t')[1]]
}
return inputs

5
modelscope/pipelines/cv/general_recognition_pipeline.py
View File

@@ -114,9 +114,8 @@ class GeneralRecognitionPipeline(Pipeline):
label_mapping = f.readlines()
score = torch.max(inputs['outputs'])
inputs = {
OutputKeys.SCORES:
score.item(),
OutputKeys.SCORES: [score.item()],
OutputKeys.LABELS:
label_mapping[inputs['outputs'].argmax()].split('\t')[1]
[label_mapping[inputs['outputs'].argmax()].split('\t')[1]]
}
return inputs