Ultralytics YOLO11来啦!更快!更强!

简介: YOLO(You Only Look Once)是一种流行的物体检测和图像分割模型,由华盛顿大学的 Joseph Redmon 和 Ali Farhadi 开发。

01

YOLO发展历程

YOLO(You Only Look Once)是一种流行的物体检测和图像分割模型,由华盛顿大学的 Joseph Redmon 和 Ali Farhadi 开发。

image.png

目前YOLO11提供两种许可选项,以适应不同的使用情况:

-AGPL-3.0 许可证:这种经 OSI 批准的开源许可证非常适合学生和爱好者使用,可促进开放协作和知识共享。

-企业许可证:该许可证专为商业用途设计,允许将Ultralytics 软件和人工智能模型无缝集成到商业产品和服务中,绕过了AGPL-3.0 的开源要求。


02

模型性能

Ultralytics YOLO11,作为新的SOTA模型,不仅继承了之前YOLO系列的优势,还引入了创新特性和改进,提升了性能和灵活性。它以快速、精准、易用为特点,成为处理目标检测、跟踪、实例分割、图像分类和姿态估计等多种视觉任务的理想选择。

image.png

03

最佳实践

我们使用魔搭社区免费算力(GPU)体验YOLO11,希望能够帮助您充分利用 YOLO11。

image.png

模型链接:

https://modelscope.cn/models/AI-ModelScope/YOLO11

数据集链接:

https://modelscope.cn/datasets/modelscope/coco2017val/

代码链接:

https://github.com/ultralytics/ultralytics/

技术文档:

https://docs.ultralytics.com/

环境安装

安装ultralytics并检查运行环境。

%pip install ultralytics
import ultralytics
ultralytics.checks()

image.png

模型推理

在魔搭社区下载模型并推理,YOLO11 可直接在命令行界面 (CLI) 中使用 `yolo` 命令执行各种任务和模式,并接受其他参数,例如 `imgsz=640`。

!modelscope download --model=AI-ModelScope/YOLO11 --local_dir ./ yolo11n.pt
!yolo predict model="/mnt/workspace/yolo11n.pt" source='https://ultralytics.com/images/zidane.jpg'

image.png

模型评估

下载COCO数据集,并在COCO数据集的 `val` 或 `test` 分割上验证模型的准确性。

# Download COCO val
!modelscope download --dataset="modelscope/coco2017val" --local_dir ./ coco2017val.zip
!unzip -q coco2017val.zip -d datasets && rm coco2017val.zip  # unzip
!modelscope download --dataset="modelscope/coco2017val" --local_dir ./ coco8.zip
!unzip -q coco8.zip -d datasets && rm coco8.zip  # unzip
# Validate YOLO11n on COCO8 val
!yolo val model="/mnt/workspace/yolo11n.pt" data=coco8.yaml

image.png

模型微调

image.png

图片来源:https://raw.githubusercontent.com/ultralytics/assets/

安装和使用可视化日志工具TensorBoard

#@title Select YOLO11 🚀 logger {run: 'auto'}
!pip install tensorboard -U
logger = 'TensorBoard' #@param ['Comet', 'TensorBoard']
if logger == 'Comet':
  %pip install -q comet_ml
  import comet_ml; comet_ml.init()
elif logger == 'TensorBoard':
  %load_ext tensorboard
  %tensorboard --logdir ./runs
# Train YOLO11n on COCO8 for 3 epochs
!yolo train model="/mnt/workspace/yolo11n.pt" data=coco8.yaml epochs=3 imgsz=640

模型导出

使用“format”参数将 YOLO11 模型导出为任何支持的格式,例如“format=onnx”或者“format=torchscript”。

!yolo export model="/mnt/workspace/yolo11n.pt" format=torchscript

04

python体验

YOLO11 采用了 Python 优先原则进行重新设计,从而实现无缝的 Python YOLO 体验。YOLO11 模型可以从经过训练的检查点加载,也可以从头开始创建。如下是使用python来训练、验证、预测和导出模型的示例代码。

from ultralytics import YOLO
# Load a model
model = YOLO('yolo11n.yaml')  # build a new model from scratch
model = YOLO('yolo11n.pt')  # load a pretrained model (recommended for training)
# Use the model
results = model.train(data='coco8.yaml', epochs=3)  # train the model
results = model.val()  # evaluate model performance on the validation set
results = model('https://ultralytics.com/images/bus.jpg')  # predict on an image
results = model.export(format='onnx')  # export the model to ONNX format

任务模型

YOLO11可以在多种视觉任务上训练,评估,推理和导出。

image.png

图片来源:https://raw.githubusercontent.com/ultralytics/assets/

检测任务

YOLO11检测模型没有后缀,使用默认的 YOLO11 模型

# Load YOLO11n, train it on COCO128 for 3 epochs and predict an image with it
from ultralytics import YOLO
model = YOLO('yolo11n.pt')  # load a pretrained YOLO11n detection model
model.train(data='coco8.yaml', epochs=3)  # train the model
model('https://ultralytics.com/images/bus.jpg')  # predict on an image

分割任务

# Load YOLO11n-seg, train it on COCO128-seg for 3 epochs and predict an image with it
from ultralytics import YOLO
!modelscope download --model=AI-ModelScope/YOLO11 --local_dir ./ yolo11n-seg.pt
model = YOLO('/mnt/workspace/yolo11n-seg.pt')  # load a pretrained YOLO11n segmentation model
model.train(data='coco8-seg.yaml', epochs=3)  # train the model
model('https://ultralytics.com/images/bus.jpg')  # predict on an image

分类任务

# Load YOLO11n-cls, train it on mnist160 for 3 epochs and predict an image with it
from ultralytics import YOLO
!modelscope download --model=AI-ModelScope/YOLO11 --local_dir ./ yolo11n-cls.pt
model = YOLO('/mnt/workspace/yolo11n-cls.pt')  # load a pretrained YOLO11n classification model
model.train(data='mnist160', epochs=3)  # train the model
model('https://ultralytics.com/images/bus.jpg')  # predict on an image

动作检测任务

# Load YOLO11n-pose, train it on COCO8-pose for 3 epochs and predict an image with it
from ultralytics import YOLO
!modelscope download --model=AI-ModelScope/YOLO11 --local_dir ./ yolo11n-pose.pt
model = YOLO('yolo11n-pose.pt')  # load a pretrained YOLO11n pose model
model.train(data='coco8-pose.yaml', epochs=3)  # train the model
model('https://ultralytics.com/images/bus.jpg')  # predict on an image

定向边界框 (OBB)

# Load YOLOv8n-obb, train it on DOTA8 for 3 epochs and predict an image with it
from ultralytics import YOLO
!modelscope download --model=AI-ModelScope/YOLO11 --local_dir ./ yolo11n-obb.pt
model = YOLO('yolo11n-obb.pt')  # load a pretrained YOLOv8n OBB model
model.train(data='dota8.yaml', epochs=3)  # train the model
model('https://ultralytics.com/images/bus.jpg')  # predict on an image

image.png

点击链接👇即可跳转模型~

https://modelscope.cn/models/AI-ModelScope/YOLO11?from=alizishequ__text

相关文章
|
6天前
|
机器学习/深度学习 人工智能 缓存
3天把Llama训成Mamba,性能不降,推理更快!
《Distillation and Acceleration of Hybrid Models》一文由日内瓦大学、Together AI、康奈尔大学和普林斯顿大学的研究者们共同完成,提出了一种将大型Transformer模型(如Llama)转化为线性RNN模型(如Mamba)的新方法。此方法不仅保持了模型性能,还大幅提升了推理速度。研究团队通过多阶段蒸馏及优化,结合预训练权重,在不同聊天基准测试中验证了模型的有效性。详情见:https://arxiv.org/pdf/2408.15237
20 3
|
自然语言处理 算法 数据挖掘
自蒸馏:一种简单高效的优化方式
背景知识蒸馏(knowledge distillation)指的是将预训练好的教师模型的知识通过蒸馏的方式迁移至学生模型,一般来说,教师模型会比学生模型网络容量更大,模型结构更复杂。对于学生而言,主要增益信息来自于更强的模型产出的带有更多可信信息的soft_label。例如下右图中,两个“2”对应的hard_label都是一样的,即0-9分类中,仅“2”类别对应概率为1.0,而soft_label
自蒸馏:一种简单高效的优化方式
|
3月前
|
机器学习/深度学习 算法 网络架构
【YOLOv8改进 - Backbone主干】EfficientRep:一种旨在提高硬件效率的RepVGG风格卷积神经网络架构
【YOLOv8改进 - Backbone主干】EfficientRep:一种旨在提高硬件效率的RepVGG风格卷积神经网络架构
|
3月前
|
并行计算 数据挖掘 PyTorch
【YOLOv8改进 - 特征融合】DySample :超轻量级且高效的动态上采样器
【YOLOv8改进 - 特征融合】DySample :超轻量级且高效的动态上采样器
【YOLOv8改进 - 特征融合】DySample :超轻量级且高效的动态上采样器
|
2月前
|
机器学习/深度学习 人工智能 前端开发
BladeDISC 深度学习编译器问题之在动态shape下优化整体性能如何解决
BladeDISC 深度学习编译器问题之在动态shape下优化整体性能如何解决
|
4月前
|
机器学习/深度学习 数据采集 人工智能
【机器学习】CLIP模型在有限计算资源下的性能探究:从数据、架构到训练策略
【机器学习】CLIP模型在有限计算资源下的性能探究:从数据、架构到训练策略
286 0
|
5月前
|
机器学习/深度学习 计算机视觉
【YOLOv8改进】 YOLOv8 更换骨干网络之GhostNetV2 长距离注意力机制增强廉价操作,构建更强端侧轻量型骨干 (论文笔记+引入代码)
该专栏聚焦YOLO目标检测的创新改进与实战,介绍了轻量级CNNs和注意力机制在移动设备上的应用。文章提出了一种名为GhostNetV2的新架构,结合了硬件友好的DFC注意力机制,强化了特征表达能力和全局信息捕获,同时保持低计算成本和高效推理。GhostNetV2在ImageNet上以167M FLOPs达到75.3%的top-1准确率,优于同类模型。创新点包括DFC注意力、模型结构优化和效率提升。源代码可在GitHub和MindSpore平台上找到。此外,还提到了YOLOv8的相关实现和任务配置。
|
5月前
|
机器学习/深度学习 人工智能 算法
提升深度学习模型性能的优化技术
【5月更文挑战第29天】 随着人工智能领域的快速发展,尤其是深度学习技术的广泛应用,如何有效提升模型的性能成为了研究热点。本文将探讨几种用于提高深度学习模型性能的优化技术,包括数据增强、正则化方法、网络结构调整以及超参数优化等。通过对比实验和理论分析,展示了这些技术在实际应用中的有效性和局限性,并提出了未来可能的研究方向。
|
5月前
|
机器学习/深度学习 存储 TensorFlow
YOLOv3模型在不同硬件平台上的性能表现有何差异?
YOLOv3模型在不同硬件平台上的性能表现有何差异?
|
机器学习/深度学习 算法 自动驾驶
YOLOX-PAI:加速YOLOX,比YOLOV6更快更强
本文,我们将逐一介绍所探索的相关改进与消融实验结果,如何基于PAI-EasyCV使用PAI-Blade优化模型推理过程,及如何使用PAI-EasyCV进行模型训练、验证、部署和端到端推理。欢迎大家关注和使用PAI-EasyCV和PAI-Blade,进行简单高效的视觉算法开发及部署任务。