Browse Source

[to #42322933]cv/tinynas/classification2

fix to the output form to list
调整输出格式,改为 list 格式
        Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9684053
master
xiachen.wyh yingda.chen 3 years ago
parent
commit
5cec664b5d
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      modelscope/pipelines/cv/tinynas_classification_pipeline.py

+ 2
- 2
modelscope/pipelines/cv/tinynas_classification_pipeline.py View File

@@ -90,7 +90,7 @@ class TinynasClassificationPipeline(Pipeline):
output_prob = torch.nn.functional.softmax(inputs['outputs'], dim=-1)
score = torch.max(output_prob)
output_dict = {
OutputKeys.SCORES: score.item(),
OutputKeys.LABELS: label_dict[inputs['outputs'].argmax().item()]
OutputKeys.SCORES: [score.item()],
OutputKeys.LABELS: [label_dict[inputs['outputs'].argmax().item()]]
}
return output_dict

Loading…
Cancel
Save