首页 > 其他分享 >lightweight处理关键点。

lightweight处理关键点。

时间:2023-02-04 13:33:06浏览次数:32  
标签:map 处理 keypoints converted lightweight reorder 关键点

ConvertKeypoints()交换关键点顺序,剩下其他代码并未交换顺序

def _convert(self, keypoints, w, h):
# Nose, Neck, R hand, L hand, R leg, L leg, Eyes, Ears
reorder_map = [1, 7, 9, 11, 6, 8, 10, 13, 15, 17, 12, 14, 16, 3, 2, 5, 4]
converted_keypoints = list(keypoints[i - 1] for i in reorder_map)
converted_keypoints.insert(1, [(keypoints[5][0] + keypoints[6][0]) / 2,
(keypoints[5][1] + keypoints[6][1]) / 2, 0]) # Add neck as a mean of shoulders
if keypoints[5][2] == 2 or keypoints[6][2] == 2:
converted_keypoints[1][2] = 2
elif keypoints[5][2] == 1 and keypoints[6][2] == 1:
converted_keypoints[1][2] = 1
if (converted_keypoints[1][0] < 0
or converted_keypoints[1][0] >= w
or converted_keypoints[1][1] < 0
or converted_keypoints[1][1] >= h):
converted_keypoints[1][2] = 2
return converted_keypoints

标签:map,处理,keypoints,converted,lightweight,reorder,关键点
From: https://www.cnblogs.com/hahaah/p/17091323.html

相关文章