4 个月前

基于Transformer的场景识别

基于Transformer的场景识别

摘要

地面情境识别(GSR)不仅对显著动作(动词)进行分类,还预测与语义角色相关的实体(名词)及其在给定图像中的位置。受Transformer在视觉任务中取得的显著成功启发,我们提出了一种基于Transformer编码器-解码器架构的GSR模型。该模型的注意力机制通过有效捕捉图像的高层次语义特征,实现了精确的动作分类,并且使模型能够灵活处理实体之间复杂且依赖于图像的关系,从而提高了名词分类和定位的准确性。我们的模型是首个应用于GSR的Transformer架构,在SWiG基准测试的每一项评估指标上均达到了当前最佳水平。我们的代码已发布在 https://github.com/jhcho99/gsrtr 。

代码仓库

jhcho99/gsrtr
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
grounded-situation-recognition-on-swigGSRTR
Top-1 Verb: 40.63
Top-1 Verb u0026 Grounded-Value: 25.49
Top-1 Verb u0026 Value: 32.15
Top-5 Verbs: 69.81
Top-5 Verbs u0026 Grounded-Value: 42.5
Top-5 Verbs u0026 Value: 54.13
situation-recognition-on-imsituGSRTR
Top-1 Verb: 40.63
Top-1 Verb u0026 Value: 32.15
Top-5 Verbs: 69.81
Top-5 Verbs u0026 Value: 54.13

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于Transformer的场景识别 | 论文 | HyperAI超神经