4 个月前

基于注意力机制的视觉关键词定位

基于注意力机制的视觉关键词定位

摘要

本文研究了在无声视频序列中识别口语关键词的任务,也称为视觉关键词识别。为此,我们探讨了基于Transformer的模型,该模型接收两个输入流:视频的视觉编码和关键词的音素编码,并输出关键词的时间位置(如果存在)。我们的贡献如下:(1) 我们提出了一种新颖的架构——Transpotter,该架构在视觉流和音素流之间采用了完全跨模态注意力机制;(2) 通过广泛的评估,我们证明了我们的模型在具有挑战性的LRW、LRS2、LRS3数据集上大幅优于现有的最先进视觉关键词识别和唇读方法;(3) 我们展示了我们的模型在极端条件下(如手语视频中的孤立口型)识别单词的能力。

代码仓库

prajwalkr/transpotter
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
visual-keyword-spotting-on-lrs2Transpotter
Top-1 Accuracy: 65
Top-5 Accuracy: 87.1
mAP: 69.2
mAP IOU@0.5: 68.3
visual-keyword-spotting-on-lrs3-tedTranspotter
Top-1 Accuracy: 52
Top-5 Accuracy: 77.1
mAP: 55.4
mAP IOU@0.5: 53.6
visual-keyword-spotting-on-lrwTranspotter
Top-1 Accuracy: 85.8
Top-5 Accuracy: 99.6
mAP: 64.1

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于注意力机制的视觉关键词定位 | 论文 | HyperAI超神经