Segment Anything 2项目新增对MPS和CPU设备的支持

Segment Anything 2(SAM2)作为Facebook Research推出的先进图像分割模型，近期迎来了一个重要更新——正式支持苹果M系列芯片的Metal Performance Shaders(MPS)以及CPU设备。这一更新解决了Mac用户和没有NVIDIA GPU设备用户的使用难题。

技术背景

在计算机视觉领域，图像分割模型通常需要强大的GPU加速计算。传统上，这类模型主要依赖NVIDIA的CUDA架构进行加速。然而，随着苹果自研芯片的普及，越来越多的开发者希望在Mac设备上直接运行这些模型。

MPS是苹果提供的Metal框架中的高性能计算组件，专门为苹果芯片优化。它能够充分利用M系列芯片的统一内存架构和强大的神经网络引擎，为深度学习推理提供接近CUDA的性能。

更新内容

此次更新主要涉及以下方面：

1. 设备兼容性扩展：模型现在可以自动检测并适配MPS设备(Mac)、CUDA设备(NVIDIA GPU)或CPU设备
2. 性能优化：针对MPS设备进行了专门的优化，确保在苹果芯片上获得最佳性能
3. 示例更新：项目中的Jupyter Notebook示例已更新，支持在不同设备上运行

使用指南

对于Mac用户，现在可以无需任何额外配置直接运行SAM2模型。系统会自动检测并使用MPS加速。对于没有GPU设备的用户，模型也可以回退到CPU模式运行，虽然速度会有所下降，但保证了功能的可用性。

技术意义

这一更新具有多重意义：

• 降低了使用门槛：使更多没有高端NVIDIA GPU设备的开发者能够体验SAM2的强大功能
• 生态扩展：完善了苹果生态下的深度学习工具链
• 研究便利性：研究人员可以在更多设备上进行实验和原型开发

未来展望

随着异构计算的发展，支持多种计算后端将成为深度学习框架的标配。SAM2的这一更新也反映了这一趋势，为后续更多设备支持奠定了基础。

对于想要在Mac设备上体验先进图像分割技术的开发者来说，这无疑是一个令人振奋的消息。现在，M系列Mac用户可以直接本地运行SAM2，无需依赖云服务或其他硬件设备。