arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2312.05760cs.CV

RepViT-SAM:迈向实时分割一切

RepViT-SAM: Towards Real-Time Segmenting Anything

Tsinghua University(清华大学);The University of Sheffield(谢菲尔德大学)

Ao Wang, Hui Chen, Zijia Lin, Jungong Han, Guiguang Ding

AI总结:

本文提出RepViT-SAM,用移动端高效RepViT替换SAM的重量级图像编码器,在零样本分割性能和推理速度上均优于MobileSAM。

摘要:

Segment Anything Model(SAM,分割一切模型)近期在多种计算机视觉任务上展现出令人瞩目的零样本迁移性能。然而,其高昂的计算成本仍令实际应用望而却步。MobileSAM通过采用蒸馏,用TinyViT替换SAM中的重量级图像编码器,从而显著降低了计算需求。但由于自注意力机制会带来大量内存和计算开销,其在资源受限移动设备上的部署仍面临挑战。近期,RepViT通过将ViT的高效架构设计融入CNN,在移动设备上实现了最先进的性能与延迟权衡。本文为在移动设备上实现实时分割一切,沿用MobileSAM的思路,将SAM中的重量级图像编码器替换为RepViT模型,最终得到RepViT-SAM模型。大量实验表明,RepViT-SAM相比MobileSAM具有显著更优的零样本迁移能力,同时推理速度提升近10倍。代码和模型可在https://github.com/THU-MIG/RepViT获取。

英文摘要:

Segment Anything Model (SAM) has shown impressive zero-shot transfer performance for various computer vision tasks recently. However, its heavy computation costs remain daunting for practical applications. MobileSAM proposes to replace the heavyweight image encoder in SAM with TinyViT by employing distillation, which results in a significant reduction in computational requirements. However, its deployment on resource-constrained mobile devices still encounters challenges due to the substantial memory and computational overhead caused by self-attention mechanisms. Recently, RepViT achieves the state-of-the-art performance and latency trade-off on mobile devices by incorporating efficient architectural designs of ViTs into CNNs. Here, to achieve real-time segmenting anything on mobile devices, following MobileSAM, we replace the heavyweight image encoder in SAM with RepViT model, ending up with the RepViT-SAM model. Extensive experiments show that RepViT-SAM can enjoy significantly better zero-shot transfer capability than MobileSAM, along with nearly $10\times$ faster inference speed. The code and models are available at \url{https://github.com/THU-MIG/RepViT}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑