跳至主要內容

20.yoloV8

约 291 字小于 1 分钟

简介

YOLO官网open in new window

YOLOv8基于深度学习和计算机视觉的前沿进展, 提供了无与伦比的速度和准确性表现。 它的精简设计使其适用于各种应用, 并且可以轻松适应不同的硬件平台,从边缘设备到云API

python 环境

安装

pip install ultralytics

pip install torch torchvision numpy matplotlib polars pyyaml pillow psutil requests scipy ultralytics-thop

pip install opencv-python-headless

命令使用以下语法:

yolo 任务 模式 参数

其中   任务(可选)是[detect, segment, classify]中的一个
        模式(必需)是[train, val, predict, export, track]中的一个
        参数(可选)是任意数量的自定义“arg=value”对,如“imgsz=320”,可覆盖默认值。

官网代码示例

from ultralytics import YOLO

# 从头开始创建一个新的YOLO模型
model = YOLO('yolov8n.yaml')

# 加载预训练的YOLO模型(推荐用于训练)
model = YOLO('yolov8n.pt')

# 使用“coco128.yaml”数据集训练模型3个周期
results = model.train(data='coco128.yaml', epochs=3)

# 评估模型在验证集上的性能
results = model.val()

# 使用模型对图片进行目标检测
results = model('https://ultralytics.com/images/bus.jpg')

# 将模型导出为ONNX格式
success = model.export(format='onnx')