fixed all nan slice issue

2024-06-09 12:29:02 +02:00 · 2024-06-09 12:29:02 +02:00 · ac078342c8
commit ac078342c8
parent cd66285690
1 changed files with 3 additions and 0 deletions
--- a/Pose2Sim/common.py
+++ b/Pose2Sim/common.py
@ -229,6 +229,9 @@ def euclidean_distance(q1, q2):
    q1 = np.array(q1)
    q2 = np.array(q2)
    dist = q2 - q1
+    if np.isnan(dist).all():
+        dist =  np.empty_like(dist)
+        dist[...] = np.inf
    
    euc_dist = np.sqrt(np.nansum( [d**2 for d in dist]))