使用 pip 安装 Ultralytics 软件包非常简单:
pip install ultralytics要安装最新开发版本,请直接从 GitHub 仓库安装:
pip install git+https://github.com/ultralytics/ultralytics.git详细的安装说明请参阅快速入门指南。
要训练自定义 YOLO 模型:
-
按照 YOLO 格式准备数据集(图像及对应的标签 txt 文件)。
-
创建一个描述数据集结构和类别的 YAML 文件(参见数据集 YAML 示例)。
-
使用以下 Python 代码开始训练:
from ultralytics import YOLO # Load a model model = YOLO("yolo26n.yaml") # build a new model from scratch model = YOLO("yolo26n.pt") # load a pretrained model (recommended for training) # Train the model results = model.train(data="path/to/your/data.yaml", epochs=100, imgsz=640)
如需更深入的指南,包括数据准备和高级训练选项,请参阅完整的训练指南。
-
Ultralytics 针对各种任务提供了多种预训练模型:
- 目标检测:YOLO26n、YOLO26s、YOLO26m、YOLO26l、YOLO26x
- 实例分割:YOLO26n-seg、YOLO26s-seg、YOLO26m-seg、YOLO26l-seg、YOLO26x-seg
- 语义分割:YOLO26n-sem、YOLO26s-sem、YOLO26m-sem、YOLO26l-sem、YOLO26x-sem
- 深度估计:YOLO26n-depth、YOLO26s-depth、YOLO26m-depth、YOLO26l-depth、YOLO26x-depth
- 分类:YOLO26n-cls、YOLO26s-cls、YOLO26m-cls、YOLO26l-cls、YOLO26x-cls
- 姿态估计:YOLO26n-pose、YOLO26s-pose、YOLO26m-pose、YOLO26l-pose、YOLO26x-pose
- 有向检测(OBB):YOLO26n-obb、YOLO26s-obb、YOLO26m-obb、YOLO26l-obb、YOLO26x-obb
这些模型在大小和复杂度方面各不相同,在速度和准确率之间提供不同的权衡。浏览完整的预训练模型系列,找到最适合你项目的模型。
要使用训练好的模型执行推理:
from ultralytics import YOLO # Load a model model = YOLO("path/to/your/model.pt") # Perform inference results = model("path/to/image.jpg") # Process results for r in results: print(r.boxes) # print bbox predictions print(r.masks) # print mask predictions print(r.probs) # print class probabilities如需高级推理选项,包括批处理和视频推理,请参阅详细的预测指南。
当然可以!Ultralytics 模型支持在各种平台上灵活部署:
- 边缘设备:使用 TensorRT、ONNX 或 OpenVINO 优化 NVIDIA Jetson 或 Raspberry Pi 等设备上的推理。
- 移动端:将模型转换为 LiteRT 或 Core ML,部署到 Android 或 iOS 设备上。
- 云端:利用 TensorFlow Serving 或 TorchServe 等框架实现可扩展的云端部署。
- Web:使用 ONNX Runtime Web 在浏览器内实现推理。
Ultralytics 提供导出函数,可将模型转换为各种格式以供部署。探索丰富的部署选项,为你的使用场景找到最佳方案。
主要区别包括:
- 端到端无 NMS 推理:Yolo26 提供带有
nms=False的无 NMS 推理。端到端无 NMS 推理的默认一对多路径使用 NMS 来保证准确性。 - 移除 DFL:YOLO26 移除了 Distribution Focal Loss 模块,简化了导出流程,并提高了与边缘设备和低功耗设备的兼容性。
- MuSGD 优化器:SGD 与 Muon 的混合优化器(受 Moonshot AI 的 Kimi K2 启发),可实现更稳定的训练和更快的收敛。
- CPU 性能:在 CPU ONNX 推理(Intel Xeon CPU @ 2.00 GHz)中,YOLO26n 的运行速度最高比 YOLO11n 快 43%,非常适合没有 GPU 的设备。
- 任务专项优化:通过语义损失和多尺度原型增强分割,通过 RLE 提升姿态估计精度,并通过角度损失改进 OBB 解码。
- 任务:两种模型都支持目标检测和实例分割,而 YOLO26 还增加了用于密集像素级预测的语义分割和用于逐像素深度预测的单目深度估计。两种模型还在统一框架中支持分类、姿态估计和有向目标检测(OBB)。
如需深入比较功能和性能指标,请访问 YOLO26 文档页面。
- 端到端无 NMS 推理:Yolo26 提供带有
为 Ultralytics 做贡献是改进项目和提升技能的好方法。你可以按以下方式参与:
- 在 GitHub 上复刻 Ultralytics 仓库。
- 为你的功能或错误修复创建新分支。
- 完成修改并确保所有测试通过。
- 提交拉取请求,并清晰说明你的修改。
- 参与代码审查流程。
你还可以通过报告错误、提出功能建议或改进文档来做贡献。详细指南和最佳实践请参阅贡献指南。
Ultralytics 提供了丰富的资源,帮助你入门并熟练掌握其工具:
- 📚 官方文档:全面的指南、API 参考和最佳实践。
- 💻 GitHub 仓库:源代码、示例脚本和社区贡献。
- ✍️ Ultralytics 博客:深度文章、使用案例和技术洞察。
- 💬 社区论坛:与其他用户交流、提问并分享你的经验。
- 🎥 YouTube 频道:涵盖各种 Ultralytics 主题的视频教程、演示和网络研讨会。
这些资源提供了使用 Ultralytics 模型完成各种任务的代码示例、真实应用案例和分步指南。
如果你需要进一步帮助,请查阅 Ultralytics 文档,或通过 GitHub Issues 或官方讨论论坛联系社区。