From d2a6845c64d9643a4ce19b1f0d132582203c961c Mon Sep 17 00:00:00 2001 From: Kevin Wang <34083603+KevinNuNu@users.noreply.github.com> Date: Tue, 27 Dec 2022 10:29:22 +0800 Subject: [PATCH] [Fix] negative number encountered in sqrt when compute distances from points to a line (#1640) --- mmocr/models/textdet/module_losses/db_module_loss.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/mmocr/models/textdet/module_losses/db_module_loss.py b/mmocr/models/textdet/module_losses/db_module_loss.py index 0c14ef2b..ba848731 100644 --- a/mmocr/models/textdet/module_losses/db_module_loss.py +++ b/mmocr/models/textdet/module_losses/db_module_loss.py @@ -286,6 +286,8 @@ class DBModuleLoss(SegBasedModuleLoss): neg_cos_c = ( (c_square - a_square - b_square) / (np.finfo(np.float32).eps + 2 * np.sqrt(a_square * b_square))) + # clip -cosC value to [-1, 1] + neg_cos_c = np.clip(neg_cos_c, -1.0, 1.0) # sinC^2=1-cosC^2 square_sin = 1 - np.square(neg_cos_c) square_sin = np.nan_to_num(square_sin)