Update viz_utils.py

viz_utils.py

@@ -66,6 +66,10 @@ def draw_keypoints_on_image(image,
     # draw ellipses around keypoints
     for i, (keypoint_x, keypoint_y) in enumerate(zip(keypoints_x, keypoints_y)):
         round_fill = list(cm.viridis(norm(colores[i]),bytes=True))#[round(num*255) for num in list(cmap(i))[:3]] #check!
+        # handling potential nans in the keypoints
+        if np.isnan(keypoint_x).any():
+            continue
+        
         if np.isnan(alpha[i]) == False : 
             round_fill[3] = round(alpha[i] *255)
         #print(round_fill)