tf-openpose-estimation得到关节点的坐标

喜欢ヅ旅行 2022-02-20 04:57 181阅读 0赞

[从这可以找到答案][Link 1]

从github的Issue中得到了一种解决办法，需要修改一下tf-pose中的estimator.py文件，其中TfPoseEstimator类中有一个函数为draw\_humans，将在其中添加一个list来保存点的坐标即可。需要添加以下三行代码，其中，flat就是存放坐标的list。

下图是作者给出的关键点的序列，得到的数组当中的下标序列与此图一致，一共18个关键点，数组大小36，下标0和1表示第0个关键点的坐标，依次类推。。

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0hvcml6b25odWk_size_16_color_FFFFFF_t_70][]

flat = [0.0 for i in range(36)]
    ....
    
    flat[i*2] = center[0]
    flat[i*2+1] = center[1]

修改完毕后为该函数增加一个返回值或是在函数中直接输出，更改后的代码如下

def draw_humans(npimg, humans, imgcopy=False):
            if imgcopy:
                npimg = np.copy(npimg)
            image_h, image_w = npimg.shape[:2]
            centers = {}
            #添加一个数组记录坐标  作为返回值返回
            flat = [0.0 for i in range(36)]
            for human in humans:
                # draw point
                for i in range(common.CocoPart.Background.value):
                    if i not in human.body_parts.keys():
                        continue
    
                    body_part = human.body_parts[i]
                    center = (int(body_part.x * image_w + 0.5), int(body_part.y * image_h + 0.5))
                    centers[i] = center
                    #自行添加   将每个关节的坐标点记录在flat中
                    flat[i*2] = center[0]
                    flat[i*2+1] = center[1]
                    cv2.circle(npimg, center, 3, common.CocoColors[i], thickness=3, lineType=8, shift=0)
    
                # draw line
                for pair_order, pair in enumerate(common.CocoPairsRender):
                    if pair[0] not in human.body_parts.keys() or pair[1] not in human.body_parts.keys():
                        continue
    
                    # npimg = cv2.line(npimg, centers[pair[0]], centers[pair[1]], common.CocoColors[pair_order], 3)
                    cv2.line(npimg, centers[pair[0]], centers[pair[1]], common.CocoColors[pair_order], 3)
            #print(flat)
            return npimg, flat

[Link 1]: https://github.com/ildoonet/tf-pose-estimation/issues?page=1&q=is%3Aissue+is%3Aopen
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0hvcml6b25odWk_size_16_color_FFFFFF_t_70]: /images/20220220/39251ca762ab4685b01852b8ac709ff8.png