arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2303.02936cs.CV

UniHCP:一种用于人类中心感知的统一模型

UniHCP: A Unified Model for Human-Centric Perceptions

The University of Sydney(悉尼大学);Zhejiang University(浙江大学);Shanghai AI Laboratory(上海人工智能实验室);SenseTime Research(商汤科技研究院);Qing Yuan Research Institute, Shanghai Jiao Tong University(上海交通大学清源研究院)

Yuanzheng Ci, Yizhou Wang, Meilin Chen, Shixiang Tang, Lei Bai, Feng Zhu, Rui Zhao, Fengwei Yu, Donglian Qi, Wanli Ouyang

AI总结:

提出统一模型UniHCP,以极简端到端方式联合33个人类中心数据集,在姿态估计、人体解析、行人检测与重识别等任务上超越专用模型,取得多项新最先进结果。

摘要:

人类中心感知(例如姿态估计、人体解析、行人检测、行人重识别等)在视觉模型的工业应用中扮演着关键角色。虽然特定的人类中心任务有其各自关注的语义方面,但它们共享相同的人体底层语义结构。然而,很少有工作尝试利用这种同质性并设计一个通用模型来处理人类中心任务。在这项工作中,我们重新审视了广泛的人类中心任务,并以极简方式将它们统一起来。我们提出了UniHCP,一种用于人类中心感知的统一模型,它通过简化的端到端方式和普通视觉变换器架构统一了广泛的人类中心任务。通过在33个人类中心数据集上进行大规模联合训练,UniHCP可以通过直接评估在多个域内和下游任务上超越强基线。当适应特定任务时,UniHCP在广泛的人类中心任务上取得了新的最先进结果,例如在CIHP上人体解析的69.8 mIoU,在PA-100K上属性预测的86.18 mA,在Market1501上行人重识别的90.3 mAP,以及在CrowdHuman上行人检测的85.8 JI,其性能优于为每个任务专门定制的模型。

英文摘要:

Human-centric perceptions (e.g., pose estimation, human parsing, pedestrian detection, person re-identification, etc.) play a key role in industrial applications of visual models. While specific human-centric tasks have their own relevant semantic aspect to focus on, they also share the same underlying semantic structure of the human body. However, few works have attempted to exploit such homogeneity and design a general-propose model for human-centric tasks. In this work, we revisit a broad range of human-centric tasks and unify them in a minimalist manner. We propose UniHCP, a Unified Model for Human-Centric Perceptions, which unifies a wide range of human-centric tasks in a simplified end-to-end manner with the plain vision transformer architecture. With large-scale joint training on 33 human-centric datasets, UniHCP can outperform strong baselines on several in-domain and downstream tasks by direct evaluation. When adapted to a specific task, UniHCP achieves new SOTAs on a wide range of human-centric tasks, e.g., 69.8 mIoU on CIHP for human parsing, 86.18 mA on PA-100K for attribute prediction, 90.3 mAP on Market1501 for ReID, and 85.8 JI on CrowdHuman for pedestrian detection, performing better than specialized models tailored for each task.

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

↑