YOLO-NAS

概述

YOLO-NAS 由 Deci AI 开发，是一款突破性的目标检测基础模型。它是先进的 神经架构搜索 技术的产物，经过精心设计，旨在解决以往 YOLO 模型的局限性。凭借在量化支持和 准确率-延迟权衡方面的重大改进，YOLO-NAS 代表了目标检测领域的一次重大飞跃。

YOLO-NAS 概览。 YOLO-NAS 采用了量化感知模块和选择性量化，以实现最佳性能。当该模型转换为其 INT8 量化版本时，精度损失极小，这比其他模型有了显著改进。这些进步造就了一种具有前所未有的目标检测能力和出色性能的卓越架构。

主要特性

• 量化友好的基础模块： YOLO-NAS 引入了一种新的基础模块，该模块对量化非常友好，解决了以往 YOLO 模型的一个显著局限性。
• 复杂的训练和量化： YOLO-NAS 利用先进的训练方案和训练后量化来增强性能。
• AutoNAC 优化和预训练： YOLO-NAS 使用 AutoNAC 优化，并在 COCO、Objects365 和 Roboflow 100 等著名数据集上进行了预训练。这种预训练使其非常适合生产环境中的下游目标检测任务。

预训练模型

通过 Ultralytics 提供的预训练 YOLO-NAS 模型，体验下一代目标检测的强大功能。这些模型旨在在速度和准确率方面提供顶级的性能。你可以从多种选项中进行选择，以满足你的特定需求：

每个模型变体旨在提供 平均准确率 (mAP) 和延迟之间的平衡，帮助你优化目标检测任务的性能和速度。

使用示例

Ultralytics 使得通过我们的 ultralytics Python 包将 YOLO-NAS 模型集成到你的 Python 应用程序中变得简单。该包提供了一个用户友好的 Python API 来简化流程。

以下示例展示了如何将 YOLO-NAS 模型与 ultralytics 包结合使用进行推理和验证：

推理和验证示例

在此示例中，我们在 COCO8 数据集上验证 YOLO-NAS-s。

from ultralytics import NAS

# Load a COCO-pretrained YOLO-NAS-s model
model = NAS("yolo_nas_s.pt")

# Display model information (optional)
model.info()

# Validate the model on the COCO8 example dataset
results = model.val(data="coco8.yaml")

# Run inference with the YOLO-NAS-s model on the 'bus.jpg' image
results = model("path/to/bus.jpg")

支持的任务和模式

我们提供三种 YOLO-NAS 模型变体：小号 (s)、中号 (m) 和大号 (l)。每种变体都旨在满足不同的计算和性能需求：

• YOLO-NAS-s：针对计算资源有限但效率至关重要的环境进行了优化。
• YOLO-NAS-m：提供了一种平衡的方法，适用于具有更高准确率的通用 目标检测。
• YOLO-NAS-l：专为需要最高准确率且计算资源限制较小的场景而设计。

以下是每个模型的详细概览，包括其预训练权重的链接、支持的任务以及与不同操作模式的兼容性。

引文与致谢

如果你在研究或开发工作中使用了 YOLO-NAS，请引用 SuperGradients：

@misc{supergradients,
      doi = {10.5281/ZENODO.7789328},
      url = {https://zenodo.org/records/7789328},
      author = {Aharon,  Shay and {Louis-Dupont} and {Ofri Masad} and Yurkova,  Kate and {Lotem Fridman} and {Lkdci} and Khvedchenya,  Eugene and Rubin,  Ran and Bagrov,  Natan and Tymchenko,  Borys and Keren,  Tomer and Zhilko,  Alexander and {Eran-Deci}},
      title = {Super-Gradients},
      publisher = {GitHub},
      journal = {GitHub repository},
      year = {2021},
}

我们向 Deci AI 的 SuperGradients 团队表示感谢，感谢他们为 计算机视觉 社区创建和维护这一宝贵资源所做的努力。我们相信，凭借其创新的架构和卓越的目标检测能力，YOLO-NAS 将成为开发人员和研究人员的重要工具。

常见问题 (FAQ)

什么是 YOLO-NAS，它比以往的 YOLO 模型有什么改进？

YOLO-NAS 由 Deci AI 开发，是一款利用先进的神经架构搜索 (NAS) 技术的最先进目标检测模型。它通过引入量化友好的基础模块和复杂的训练方案等功能，解决了以往 YOLO 模型的局限性。这带来了显著的性能提升，特别是在计算资源有限的环境中。YOLO-NAS 还支持量化，即使在转换为其 INT8 版本时也能保持高准确率，增强了其对生产环境的适应性。欲了解更多详情，请参阅 概览 部分。

我如何将 YOLO-NAS 模型集成到我的 Python 应用程序中？

你可以使用 ultralytics 包轻松地将 YOLO-NAS 模型集成到你的 Python 应用程序中。以下是如何加载预训练 YOLO-NAS 模型并执行推理的简单示例：

from ultralytics import NAS

# Load a COCO-pretrained YOLO-NAS-s model
model = NAS("yolo_nas_s.pt")

# Validate the model on the COCO8 example dataset
results = model.val(data="coco8.yaml")

# Run inference with the YOLO-NAS-s model on the 'bus.jpg' image
results = model("path/to/bus.jpg")

更多信息，请参考 推理和验证示例。

YOLO-NAS 的主要功能是什么，为什么我应该考虑使用它？

YOLO-NAS 引入了几个关键功能，使其成为目标检测任务的绝佳选择：

• 量化友好的基础模块： 增强的架构提高了模型性能，且量化后的 精度 下降极小。
• 复杂的训练和量化： 采用先进的训练方案和训练后量化技术。
• AutoNAC 优化和预训练： 利用 AutoNAC 优化，并在 COCO、Objects365 和 Roboflow 100 等著名数据集上进行预训练。

这些功能有助于其实现高准确率、高效性能，并适合在生产环境中部署。在 主要功能 部分了解更多信息。

YOLO-NAS 模型支持哪些任务和模式？

YOLO-NAS 模型支持各种目标检测任务和模式，如推理、验证和导出。它们不支持训练。支持的模型包括 YOLO-NAS-s、YOLO-NAS-m 和 YOLO-NAS-l，每种模型都针对不同的计算能力和性能需求进行了调整。有关详细概览，请参考 支持的任务和模式 部分。

是否有预训练的 YOLO-NAS 模型可用，我该如何获取它们？

是的，Ultralytics 提供了你可以直接访问的预训练 YOLO-NAS 模型。这些模型在 COCO 等数据集上进行了预训练，确保了速度和准确率方面的高性能。你可以使用 预训练模型 部分中提供的链接下载这些模型。以下是一些示例：

• YOLO-NAS-s
• YOLO-NAS-m
• YOLO-NAS-l