2 个月前

SipMask:快速图像和视频实例分割中的空间信息保持

Jiale Cao; Rao Muhammad Anwer; Hisham Cholakkal; Fahad Shahbaz Khan; Yanwei Pang; Ling Shao
SipMask:快速图像和视频实例分割中的空间信息保持
摘要

单阶段实例分割方法由于其速度和简洁性最近受到了广泛关注,但在准确性方面仍落后于两阶段方法。我们提出了一种快速的单阶段实例分割方法,称为SipMask,该方法通过将检测到的边界框内的实例掩码预测划分为不同的子区域来保留实例特定的空间信息。我们的主要贡献是一种新颖的轻量级空间保留(SP)模块,该模块为边界框内的每个子区域生成一组独立的空间系数,从而提高了掩码预测的准确性。此外,它还能够精确地划分相邻的空间实例。为了更好地关联掩码预测与目标检测,我们引入了掩码对齐加权损失和特征对齐方案。在COCO测试开发集上,我们的SipMask优于现有的单阶段方法。与最先进的单阶段TensorMask相比,SipMask在掩码AP上获得了1.0%的绝对提升,同时提供了四倍的速度提升。在实时性能方面,在类似设置下,SipMask在掩码AP上比YOLACT高出3.0%,而在Titan Xp上的运行速度相当。我们还在YouTube-VIS数据集上评估了SipMask用于实时视频实例分割的效果,取得了令人鼓舞的结果。源代码可在https://github.com/JialeCao001/SipMask 获取。