aidlab/sora_python
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

output both scores and image features

Browse Source
This commit is contained in:
pangkaicheng
2024-03-25 10:06:12 +08:00
parent 8973479756
commit 2b5dce50bb
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
app/service/outfit_matcher/outfit_evaluator.py
View File

@@ -327,9 +327,11 @@ class OutfitMaterTypeAware(OutfitMatcher):
# 输出集
outputs = [
httpclient.InferRequestedOutput("output__0", binary_data=True),
httpclient.InferRequestedOutput("output__1", binary_data=True),
]
results = client.infer(model_name="outfit_matcher_type_aware", inputs=inputs, outputs=outputs)
# 推理
# 取结果
scores = torch.from_numpy(results.as_numpy("output__0"))
return scores # Shape (N, 1)
scores = torch.from_numpy(results.as_numpy("output__0")) # Shape (N, 1)
features = torch.from_numpy(results.as_numpy("output__1")) # Shape (N, -1, 64)
return scores, features