[Bug Fix] fix cosine_similarity error in examples (PaddlePaddle#648)

cosine_similarity is error 余弦相似度计算错误 np.linalg.norm 二范数已经开过根号 Co-Authored-By: Chris Kong <chriskong93@163.com>
felixhjh · Nov 25, 2022 · ef49968 · ef49968
1 parent d98c360
commit ef49968
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/examples/vision/faceid/insightface/python/infer_arcface.py b/examples/vision/faceid/insightface/python/infer_arcface.py
@@ -10,7 +10,7 @@ def cosine_similarity(a, b):
     mul_a = np.linalg.norm(a, ord=2)
     mul_b = np.linalg.norm(b, ord=2)
     mul_ab = np.dot(a, b)
-    return mul_ab / (np.sqrt(mul_a) * np.sqrt(mul_b))
+    return mul_ab / (mul_a * mul_b)
 
 
 def parse_arguments():

diff --git a/examples/vision/faceid/insightface/python/infer_cosface.py b/examples/vision/faceid/insightface/python/infer_cosface.py
@@ -10,7 +10,7 @@ def cosine_similarity(a, b):
     mul_a = np.linalg.norm(a, ord=2)
     mul_b = np.linalg.norm(b, ord=2)
     mul_ab = np.dot(a, b)
-    return mul_ab / (np.sqrt(mul_a) * np.sqrt(mul_b))
+    return mul_ab / (mul_a * mul_b)
 
 
 def parse_arguments():

diff --git a/examples/vision/faceid/insightface/python/infer_partial_fc.py b/examples/vision/faceid/insightface/python/infer_partial_fc.py
@@ -10,7 +10,7 @@ def cosine_similarity(a, b):
     mul_a = np.linalg.norm(a, ord=2)
     mul_b = np.linalg.norm(b, ord=2)
     mul_ab = np.dot(a, b)
-    return mul_ab / (np.sqrt(mul_a) * np.sqrt(mul_b))
+    return mul_ab / (mul_a * mul_b)
 
 
 def parse_arguments():

diff --git a/examples/vision/faceid/insightface/python/infer_vpl.py b/examples/vision/faceid/insightface/python/infer_vpl.py
@@ -10,7 +10,7 @@ def cosine_similarity(a, b):
     mul_a = np.linalg.norm(a, ord=2)
     mul_b = np.linalg.norm(b, ord=2)
     mul_ab = np.dot(a, b)
-    return mul_ab / (np.sqrt(mul_a) * np.sqrt(mul_b))
+    return mul_ab / (mul_a * mul_b)
 
 
 def parse_arguments():