Tiny-YOLOSAM:快速混合图像分割
计算机视觉与模式识别
2025-12-30 v1
摘要
Segment Anything Model (SAM) 能够实现可提示的高质量分割,但在延迟敏感的场景中往往计算开销过大。TinySAM 是一种轻量级、从 SAM 蒸馏得到的变体,保留了强大的零样本掩码质量,但其“segment-everything”模式仍需数百个提示,且实际运行速度仍较慢。我们首先在 COCO val2017 上使用官方检查点复现 TinySAM,结果在 AP 上与报告值相差不足 0.03%,建立了可靠的实验基线。基于此,我们提出 Tiny-YOLOSAM,一种快速混合管道,利用最新的 YOLO 检测器(YOLOv12)为 TinySAM 生成显著前景对象的边界框提示,并在 YOLO 引导的掩码未覆盖区域补充稀疏点提示。实验表明,在 COCO val2017 上,该混合系统在类别无关覆盖率方面显著提升(AR 从 16.4% 提升至 77.1%,mIoU 从 19.2% 提升至 67.8%),同时将端到端运行时间从 49.20 秒/图像缩短至 10.39 秒/图像(加快 4.7 倍),在 Apple M1 Pro CPU 上实现。这些结果表明,检测器引导的提示结合有针对性的稀疏采样,是实用全场景分割中替代密集“segment-everything”提示的有效方法。
引用
@article{arxiv.2512.22193,
title = {Tiny-YOLOSAM: Fast Hybrid Image Segmentation},
author = {Kenneth Xu and Songhan Wu},
journal= {arXiv preprint arXiv:2512.22193},
year = {2025}
}
备注
7 pages, 6 figures, 5 tables. Code available at: https://github.com/Kenneth-Xu11566/tiny-yolosam