LogosIntelligenceLab/SeaFormer

[ICLR 2023 & IJCV 2025] SeaFormer: Squeeze-enhanced Axial Transformer

解决的问题

SeaFormer 解决了在移动设备上运行高性能计算机视觉模型的挑战。它提供了一种在保持高精度和低延迟之间平衡的同时,执行图像分类和语义分割等复杂视觉任务的方法。

工作原理

SeaFormer 使用了一种称为 Squeeze-enhanced Axial Transformer 的专用架构。该设计通过“压缩”信息来减少计算开销,同时利用轴向注意力机制来捕获图像中重要的空间关系,从而优化了模型处理视觉数据的方式。这使得模型在不牺牲准确识别和分割对象能力的前提下,足够轻量化以适应移动硬件。

适用人群

该项目适用于从事移动计算机视觉应用开发和研究的人员,他们需要为场景理解或图像识别等任务提供高效、高速的模型。

亮点

  • 针对移动设备进行了优化,专注于低延迟和高性能。
  • 支持图像分类和语义分割任务。
  • 提供多种模型尺寸(Tiny, Small, Base, Large)以适应不同的硬件限制。
  • 在 ImageNet-1K, ADE20K 和 Cityscapes 等标准基准测试中表现优异。

相关

  • 项目
  • 项目
  • 项目
  • 项目