4 个月前

使用注意力机制和软 argmax 进行无监督多目标分割

使用注意力机制和软 argmax 进行无监督多目标分割

摘要

我们提出了一种新的无监督对象中心表示学习及多对象检测与分割架构,该架构采用平移等变注意力机制来预测场景中对象的坐标,并为每个对象关联一个特征向量。变压器编码器负责处理遮挡和冗余检测问题,而卷积自动编码器则负责背景重建。实验结果表明,该架构在复杂的合成基准测试中显著优于现有最先进方法。

代码仓库

BrunoSauvalle/AST
官方
pytorch

基准测试

基准方法指标
unsupervised-object-segmentation-onAST
ARI-FG: 0.82
unsupervised-object-segmentation-on-1AST
ARI-FG: 0.87
unsupervised-object-segmentation-on-clevrtexAST
MSE: 167± 1
mIoU: 66.62± 0.80
unsupervised-object-segmentation-on-clevrtexAST-Seg-B3-CT
MSE: 139±7
mIoU: 79.58±0.54

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
使用注意力机制和软 argmax 进行无监督多目标分割 | 论文 | HyperAI超神经