test

app.py

@@ -8,7 +8,7 @@ import gradio as gr
 import torch
 import torchvision
 from dlclive import DLCLive, Processor
-
+import matplotlib
 from PIL import Image, ImageColor, ImageFont, ImageDraw 
 # check git lfs pull!!
 import numpy as np
@@ -62,7 +62,7 @@ def draw_keypoints_on_image(image,
     use_normalized_coordinates: if True (default), treat keypoint values as
         relative to the image.  Otherwise treat them as absolute.
 
-    alpha is the lklh (check)
+    
     """
     # get a drawing context
     draw = ImageDraw.Draw(image)  
@@ -76,14 +76,16 @@ def draw_keypoints_on_image(image,
     if use_normalized_coordinates:
         keypoints_x = tuple([im_width * x for x in keypoints_x])
         keypoints_y = tuple([im_height * y for y in keypoints_y])
-
+    
+    cmap = matplotlib.cm.get_cmap('viridis')
+    cmap2 = matplotlib.cm.get_cmap('Greys')
     # draw ellipses around keypoints
     for i, (keypoint_x, keypoint_y) in enumerate(zip(keypoints_x, keypoints_y)):
-        marker_size = alpha[i]*10
+        round_fill = [round(num*255) for num in list(cmap(i*len(keypoints_x)))[:3]]
+        round_outline = [round(num*255) for num in list(cmap2(alpha[i]))[:3]]
         draw.ellipse([(keypoint_x - marker_size, keypoint_y - marker_size),
                       (keypoint_x + marker_size, keypoint_y + marker_size)],
-                       outline=keypt_color, 
-                       fill=keypt_color)
+                       fill=tuple(round_fill), outline= tuple(round_outline))
 
         # add string labels around keypoints
         if flag_show_str_labels: