关于保存特征的一些代码

2024-03-26 18:00:35 +08:00
parent 7dce70027a
commit 88c815dd9d
16 changed files with 378 additions and 10 deletions
--- a/app/service/outfit_matcher/outfit_evaluator.py
+++ b/app/service/outfit_matcher/outfit_evaluator.py
@@ -327,9 +327,12 @@ 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, 64)
+
+        return scores, features