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

ProbPose:一种用于2D人体姿态估计的概率方法

ProbPose: A Probabilistic Approach to 2D Human Pose Estimation

Visual Recognition Group(视觉识别组);Department of Cybernetics(cybernetics系);Faculty of Electrical Engineering(电气工程学院);Czech Technical University in Prague(布拉格捷克技术大学)

Miroslav Purkrabek, Jiri Matas

AI总结:

针对现有2D人体姿态估计模型忽略图像外关键点、热图未校准的问题,提出概率方法ProbPose,配套新数据集和评估指标,提升了内外关键点定位精度与边缘鲁棒性。

摘要:

当前的人体姿态估计方法已取得显著进展,但现有最优模型忽略了图像外关键点,且使用未校准的热图作为关键点位置表示。为解决这些局限,我们提出ProbPose,它为每个关键点预测:激活窗口内每个位置存在关键点的校准概率、关键点位于窗口外的概率,以及预测的可见性。为解决图像外关键点缺乏评估协议的问题,我们引入CropCOCO数据集和Extended OKS(Ex-OKS)指标,将OKS扩展至图像外点。在COCO、CropCOCO和OCHuman上测试表明,ProbPose在图像外关键点定位方面取得显著提升,同时通过数据增强改善了图像内定位。此外,该模型提升了边界框边缘处的鲁棒性,并在关键点评估中提供了更好的灵活性。代码和模型可在https://mirapurkrabek.github.io/ProbPose/ 获取用于研究。

英文摘要:

Current Human Pose Estimation methods have achieved significant improvements. However, state-of-the-art models ignore out-of-image keypoints and use uncalibrated heatmaps as keypoint location representations. To address these limitations, we propose ProbPose, which predicts for each keypoint: a calibrated probability of keypoint presence at each location in the activation window, the probability of being outside of it, and its predicted visibility. To address the lack of evaluation protocols for out-of-image keypoints, we introduce the CropCOCO dataset and the Extended OKS (Ex-OKS) metric, which extends OKS to out-of-image points. Tested on COCO, CropCOCO, and OCHuman, ProbPose shows significant gains in out-of-image keypoint localization while also improving in-image localization through data augmentation. Additionally, the model improves robustness along the edges of the bounding box and offers better flexibility in keypoint evaluation. The code and models are available on https://mirapurkrabek.github.io/ProbPose/ for research purposes.

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

↑