群体再识别图网络

规模十倍于现有数据集

一个统一框架同时解决群体再识别与群体感知的单人再识别,利用多级注意力图神经网络充分建模群体内外上下文。

晏轶超 · 钦杰 · 倪冰冰 · 陈佳鑫 · Liu, Li · 70422345 · Zheng, Wei-Shi · 杨小康 · Shao, Ling

IEEE Transactions on Pattern Analysis and Machine Intelligence 2023

参数信息

数据集图像数量
3.8 K
标注群体数量
1.5 K
数据集规模对比
1 order of magnitude

技术优势

同时处理两种任务

统一框架基于图神经网络,同时解决群体再识别和群体感知的单人再识别,避免了为每个任务单独设计模型的麻烦。

建模群体结构

构建以群体成员为节点的上下文图,利用多级注意力机制建模群体内和群体间上下文,捕获成员间的依赖关系。

提供大规模数据集

新数据集包含超过3.8K图像和1.5K标注群体,规模比现有群体再识别数据集大一个数量级,有助于深度模型的训练。

应用场景