Browse Source

[to #42322933] fix cv/animalRecog output format

* fix output format
        Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10491722
master
yanheng.wyh yingda.chen 3 years ago
parent
commit
cf831dbf98
2 changed files with 4 additions and 6 deletions
  1. +2
    -3
      modelscope/pipelines/cv/animal_recognition_pipeline.py
  2. +2
    -3
      modelscope/pipelines/cv/general_recognition_pipeline.py

+ 2
- 3
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

+ 2
- 3
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

Loading…
Cancel
Save