From da0017691a3d398bc684754f5b001d6e6570f095 Mon Sep 17 00:00:00 2001
From: LielinJiang <50691816+LielinJiang@users.noreply.github.com>
Date: Tue, 27 Aug 2019 13:47:45 +0800
Subject: [PATCH] Fix arguement k of recall_topk method never been used (#3150)

---
 PaddleCV/metric_learning/utility.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/PaddleCV/metric_learning/utility.py b/PaddleCV/metric_learning/utility.py
index f093f56d..1834d327 100644
--- a/PaddleCV/metric_learning/utility.py
+++ b/PaddleCV/metric_learning/utility.py
@@ -87,9 +87,11 @@ def recall_topk(fea, lab, k = 1):
     sorted_index = np.argsort(d, 1)
     res = 0
     for i in range(len(fea)):
-        pred = lab[sorted_index[i][0]]
-        if lab[i] == pred:
-            res += 1.0
+        for j in range(k):
+            pred = lab[sorted_index[i][j]]
+            if lab[i] == pred:
+                res += 1.0
+                break
     res = res / len(fea)
     return res
 
-- 
GitLab