fffiloni commited on
Commit
21e82dd
·
1 Parent(s): 2e0b411

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -90,7 +90,7 @@ def predict(img):
90
  if keypoints[start_idx][2] > 0.1 and keypoints[end_idx][2] > 0.1:
91
  pt1 = (int(keypoints[start_idx][0]), int(keypoints[start_idx][1]))
92
  pt2 = (int(keypoints[end_idx][0]), int(keypoints[end_idx][1]))
93
- cv2.line(black_img, pt1, pt2, colors[start_part], thickness=2, lineType=cv2.LINE_AA)
94
 
95
  # draw circles at each keypoint
96
  for i in range(keypoints.shape[0]):
 
90
  if keypoints[start_idx][2] > 0.1 and keypoints[end_idx][2] > 0.1:
91
  pt1 = (int(keypoints[start_idx][0]), int(keypoints[start_idx][1]))
92
  pt2 = (int(keypoints[end_idx][0]), int(keypoints[end_idx][1]))
93
+ cv2.line(black_img, pt1, pt2, body_part_colors[start_part], thickness=2, lineType=cv2.LINE_AA)
94
 
95
  # draw circles at each keypoint
96
  for i in range(keypoints.shape[0]):