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

DMAT:一种用于人体去遮挡的动态掩码感知Transformer

DMAT: A Dynamic Mask-Aware Transformer for Human De-occlusion

Northwestern Polytechnical University(西北工业大学)

Guoqiang Liang, Jiahao Hu, Qingyue Wang, Shizhou Zhang

AI总结:

本文提出动态掩码感知Transformer(DMAT),通过扩展卷积头、动态多头人体掩码引导注意力及非模态损失,在AHP数据集上实现优于现有方法的人体去遮挡性能。

摘要:

人体去遮挡旨在从被遮挡的图像中推断出不可见人体部位的外观,在行人重识别、意图推断等许多人相关任务中具有重要价值。为解决该任务,本文提出了一种动态掩码感知Transformer(DMAT),它动态增强来自人体区域的信息,并削弱来自遮挡区域的信息。首先,为增强令牌表示,我们设计了一个带有扩大卷积核的扩展卷积头,该卷积头能捕获更多局部有效上下文,并减轻周围遮挡的影响。为聚焦于可见的人体部位,我们提出了一种新颖的、通过整合多个掩码的动态多头人体掩码引导注意力机制,该机制能防止被去遮挡区域同化为背景。此外,利用区域上采样策略来减轻遮挡对插值图像的影响。在模型学习过程中,开发了一种非模态损失,以进一步强调人体区域的恢复效果,同时该损失也优化了模型的收敛性。在AHP数据集上进行的大量实验表明,与近期最先进的方法相比,该方法具有优越的性能。

英文摘要:

Human de-occlusion, which aims to infer the appearance of invisible human parts from an occluded image, has great value in many human-related tasks, such as person re-id, and intention inference. To address this task, this paper proposes a dynamic mask-aware transformer (DMAT), which dynamically augments information from human regions and weakens that from occlusion. First, to enhance token representation, we design an expanded convolution head with enlarged kernels, which captures more local valid context and mitigates the influence of surrounding occlusion. To concentrate on the visible human parts, we propose a novel dynamic multi-head human-mask guided attention mechanism through integrating multiple masks, which can prevent the de-occluded regions from assimilating to the background. Besides, a region upsampling strategy is utilized to alleviate the impact of occlusion on interpolated images. During model learning, an amodal loss is developed to further emphasize the recovery effect of human regions, which also refines the model's convergence. Extensive experiments on the AHP dataset demonstrate its superior performance compared to recent state-of-the-art methods.

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

↑