arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2007.11864cs.CV
收藏

基于可微层次图的多人位姿估计

Differentiable Hierarchical Graph Grouping for Multi-Person Pose Estimation

The University of Hong Kong(香港大学);SenseTime Research(商汤科技研究院);Nanjing University(南京大学);The University of Sydney(悉尼大学)

Sheng Jin, Wentao Liu, Enze Xie, Wenhai Wang, Chen Qian, Wanli Ouyang, Ping Luo

摘要:

多人姿态估计是具有挑战性的,因为它同时定位多个人的身体关键点。以前的方法可以分为两个流,即自上而下和自下而上的方法。自上而下的方法在人体检测之后定位关键点,而自下而上的方法直接定位关键点,然后针对不同的人对它们进行聚类/分组,这通常比自上而下的方法更有效。然而,在现有的自下而上的方法中,关键点分组通常独立于关键点检测来解决,使得它们不是端到端可训练的并且具有次优性能。在本文中,我们研究了一个新的角度,人体部分分组和重新制定它作为一个图聚类任务。特别是,我们提出了一种新的可微层次图学习(HGG)方法来学习自底向上的多人姿态估计任务中的图分组。此外,HGG很容易嵌入到主流的自底向上方法中。该算法将人类关键点候选者作为图节点,在多层图神经网络模型中对关键点进行聚类。HGG的模块可以与关键点检测网络进行端到端的训练,并且能够以分层的方式监督分组过程。为了提高聚类的区分度,我们增加了一组边缘鉴别器和宏节点鉴别器。在COCO和OCHuman数据集上的大量实验表明,该方法提高了自底向上姿态估计方法的性能。

英文摘要:

Multi-person pose estimation is challenging because it localizes body keypoints for multiple persons simultaneously. Previous methods can be divided into two streams, i.e. top-down and bottom-up methods. The top-down methods localize keypoints after human detection, while the bottom-up methods localize keypoints directly and then cluster/group them for different persons, which are generally more efficient than top-down methods. However, in existing bottom-up methods, the keypoint grouping is usually solved independently from keypoint detection, making them not end-to-end trainable and have sub-optimal performance. In this paper, we investigate a new perspective of human part grouping and reformulate it as a graph clustering task. Especially, we propose a novel differentiable Hierarchical Graph Grouping (HGG) method to learn the graph grouping in bottom-up multi-person pose estimation task. Moreover, HGG is easily embedded into main-stream bottom-up methods. It takes human keypoint candidates as graph nodes and clusters keypoints in a multi-layer graph neural network model. The modules of HGG can be trained end-to-end with the keypoint detection network and is able to supervise the grouping process in a hierarchical manner. To improve the discrimination of the clustering, we add a set of edge discriminators and macro-node discriminators. Extensive experiments on both COCO and OCHuman datasets demonstrate that the proposed method improves the performance of bottom-up pose estimation methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑