Update viz_utils.py

viz_utils.py

@@ -65,6 +65,8 @@ def draw_keypoints_on_image(image,
         round_fill = list(cm.viridis(norm(colores[i]),bytes=True))#[round(num*255) for num in list(cmap(i))[:3]] #check!
         if np.isnan(alpha[i]) == False : 
             round_fill[3] = round(alpha[i] *255)
+        if np.isnan(keypoint_x).any():
+            continue
         #print(round_fill)
         #round_outline = [round(num*255) for num in list(cmap2(alpha[i]))[:3]]
         draw.ellipse([(keypoint_x - marker_size, keypoint_y - marker_size),