fffiloni commited on
Commit
dd2cb55
Β·
1 Parent(s): af92f07

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -60,6 +60,7 @@ def predict(img):
60
 
61
  # draw lines between keypoints to form a skeleton
62
  skeleton = [(0,1), (1,2), (2,3), (3,4), (1,5), (5,6), (6,7), (1,8), (8,9), (9,10), (10,11), (8,12), (12,13), (13,14), (0,15), (15,17), (0,16), (16,18)]
 
63
  for i, j in skeleton:
64
  if keypoints[i][2] < 0.1 or keypoints[j][2] < 0.1:
65
  continue
 
60
 
61
  # draw lines between keypoints to form a skeleton
62
  skeleton = [(0,1), (1,2), (2,3), (3,4), (1,5), (5,6), (6,7), (1,8), (8,9), (9,10), (10,11), (8,12), (12,13), (13,14), (0,15), (15,17), (0,16), (16,18)]
63
+
64
  for i, j in skeleton:
65
  if keypoints[i][2] < 0.1 or keypoints[j][2] < 0.1:
66
  continue