Spaces:

DeepLabCut
/

MegaDetector_DeepLabCut

Build error

sabrinabenas commited on Aug 19, 2022

Commit

3bf787d

•

1 Parent(s): d3c408d

test

Files changed (1) hide show

app.py CHANGED Viewed

@@ -79,7 +79,7 @@ def draw_keypoints_on_image(image,
     # draw ellipses around keypoints
     for i, (keypoint_x, keypoint_y) in enumerate(zip(keypoints_x, keypoints_y)):
-        marker_size = alpha[i]
         draw.ellipse([(keypoint_x - marker_size, keypoint_y - marker_size),
                       (keypoint_x + marker_size, keypoint_y + marker_size)],
                        outline=keypt_color,

     # draw ellipses around keypoints
     for i, (keypoint_x, keypoint_y) in enumerate(zip(keypoints_x, keypoints_y)):
+        marker_size = alpha[i]*10
         draw.ellipse([(keypoint_x - marker_size, keypoint_y - marker_size),
                       (keypoint_x + marker_size, keypoint_y + marker_size)],
                        outline=keypt_color,