4 个月前

SSTVOS:用于视频对象分割的稀疏时空变换器

SSTVOS:用于视频对象分割的稀疏时空变换器

摘要

本文介绍了一种基于Transformer的视频目标分割(VOS)方法。为了解决先前工作中的累积误差和可扩展性问题,我们提出了一种称为稀疏时空Transformer(SST)的可扩展端到端VOS方法。SST通过在时空特征上应用稀疏注意力机制,提取视频中每个目标的逐像素表示。我们提出的基于注意力机制的VOS公式允许模型学习关注多个帧的历史记录,并提供了适合执行解决运动分割所需的类似对应计算的归纳偏差。我们在时空域中展示了基于注意力机制的方法相对于递归网络的有效性。我们的方法在YouTube-VOS和DAVIS 2017数据集上取得了具有竞争力的结果,相比现有最佳方法,其可扩展性和对遮挡的鲁棒性得到了改善。代码可在https://github.com/dukebw/SSTVOS 获取。

代码仓库

dukebw/SSTVOS
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
semi-supervised-video-object-segmentation-on-20SSTVOS
D17 val (F): 81.4
D17 val (G): 78.4
D17 val (J): 75.4
video-object-segmentation-on-youtube-vos-1SST (Local)
Jaccard (Seen): 80.9
Jaccard (Unseen): 76.6
video-object-segmentation-on-youtube-vos-2019-2SST
Jaccard (Seen): 80.9
Jaccard (Unseen): 76.6
Mean Jaccard u0026 F-Measure: 81.8

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
SSTVOS:用于视频对象分割的稀疏时空变换器 | 论文 | HyperAI超神经