From 8e56b8105595e7beb1904d16585d79c24314264f Mon Sep 17 00:00:00 2001 From: Chris Kong <609027949@qq.com> Date: Mon, 21 Nov 2022 21:25:27 +0800 Subject: [PATCH] [Bug Fix] fix cosine_similarity error in examples (#648) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit cosine_similarity is error 余弦相似度计算错误 np.linalg.norm 二范数 已经开过根号 Co-Authored-By: Chris Kong --- examples/vision/faceid/insightface/python/infer_arcface.py | 2 +- examples/vision/faceid/insightface/python/infer_cosface.py | 2 +- examples/vision/faceid/insightface/python/infer_partial_fc.py | 2 +- examples/vision/faceid/insightface/python/infer_vpl.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/examples/vision/faceid/insightface/python/infer_arcface.py b/examples/vision/faceid/insightface/python/infer_arcface.py index a9846b4cc..a400bd32f 100644 --- 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 index 7b45f7a40..267bff37d 100644 --- 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 index b1b2f3bf1..2e503116f 100644 --- 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 index 7618913f7..dad84d647 100644 --- 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():