Labelme AI-Points 功能背后的ONNX模型：下载、配置与使用解析

---

Labelme AI-Points 功能背后的ONNX模型：下载、配置与使用解析

前言：为什么需要关注这些模型？

在使用Labelme进行图像标注时，其"AI-Points"功能（即AI多边形自动标注）能大幅提升效率。但这个功能并非凭空产生，而是依赖于底层的ONNX格式模型。许多用户在GitHub上寻找这些模型文件时，常会因为文件分散在不同仓库而绕晕。

本文将为您清晰梳理Labelme AI-Points功能所使用的ONNX模型来源、下载链接、存放配置以及技术原理，帮助您快速上手并理解其背后的机制。

核心模型一览

Labelme集成了两个主流的分割模型系列，并为其生成了轻量化的ONNX版本。

模型系列	类型/规格	编码器 (Encoder)	解码器 (Decoder)
SegmentAnything (SAM)	Speed (ViT-B)	sam_vit_b_01ec64.quantized.encoder.onnx	sam_vit_b_01ec64.quantized.decoder.onnx
SegmentAnything (SAM)	Balanced (ViT-L)	sam_vit_l_0b3195.quantized.encoder.onnx	sam_vit_l_0b3195.quantized.decoder.onnx
EfficientSAM	Ti (vitt)	efficient_sam_vitt_encoder.onnx	efficient_sam_vitt_decoder.onnx
EfficientSAM	S (vits)	efficient_sam_vits_encoder.onnx	efficient_sam_vits_decoder.onnx

如何定位这些模型（寻找路径指南）

如果未来链接有变化，或您想探索新模型，可以通过以下路径找到它们：

1. EfficientSAM 模型：

   • 在GitHub上搜索并进入仓库 labelmeai/efficient-sam。
   • 点击页面右侧的 "Releases" 标签。
   • 最新的ONNX模型通常发布在 Onnx Models 2023-12-25 这个版本中。

2. SegmentAnything (SAM) 模型：

   • 进入主仓库 wkentaro/labelme 的 "Releases" 页面。
   • 在较早期（例如第5页左右）找到 v5.2.0.post4 版本。
   • 在其发布说明中，即可找到 Segment Anything Model 相关的ONNX文件下载信息。

模型配置：让Labelme识别并加载

下载模型文件后，只需将它们放入Labelme指定的本地缓存目录，软件即可自动识别和加载。

• Windows系统 ：C:\Users\你的用户名\.cache\gdown
• Linux/macOS系统 ：~/.cache/gdown

技术解析：为什么采用ONNX格式？

采用ONNX（Open Neural Network Exchange）格式，对于Labelme这类应用有显著好处：

1. 跨框架兼容性：ONNX作为开放式格式，允许模型在PyTorch、TensorFlow等不同深度学习框架间无缝迁移，便于集成。
2. 推理加速：ONNX Runtime等执行引擎能针对不同硬件（CPU、GPU）进行深度优化，显著提升模型推理速度，这对于需要即时交互的标注工具至关重要。
3. 模型轻量化 ：Labelme提供的模型多为量化版本（文件名含quantized），在保持精度的同时，大幅减小了模型体积和内存占用，使其更适合在本地设备上运行。

原理示意：用户操作（点击、框选） -> Labelme生成提示（Points/Box） -> 输入ONNX模型（编码器+解码器） -> 输出分割掩码（Mask） -> 转换为多边形标注。

注意事项

1. 成对使用 ：每个AI模型都必须同时具备**编码器（Encoder）和 解码器（Decoder）**两个.onnx文件，缺一不可。
2. 版本支持：较新的Labelme版本（如v6.2+）能更好地支持这些模型，并增加了自动去重等功能，建议保持软件更新。
3. 自定义模型限制：目前Labelme暂不支持用户直接加载训练的自定义ONNX模型来替换这些官方模型。

总结

Labelme的AI-Points功能通过集成经过优化的SAM和EfficientSAM的ONNX模型，为用户提供了强大且高效的半自动标注体验。了解这些模型的来源、下载方式和配置方法，不仅能帮助您顺利使用该功能，也能为未来可能的二次开发或模型替换打下基础。

希望这篇博客能帮助您顺利解决模型下载和配置的问题。如果您有任何其他技术细节想探讨，欢迎进一步交流。