Python Package

Ultralytics

2024-02-01

ultralytics 是一个用于目标检测的 [[Python]] 库。

安装

Terminal window
pip install ultralytics

语法

支持在命令行中调用:

Terminal window
yolo TASK MODE ARGS
  • TASK:
    • detect: 目标检测
    • segment: 语义分割
    • classify: 图像分类
    • pose: 姿态估计
    • obb: 方向包围框
  • MODE:
    • train: 训练
    • val: 验证
    • predict: 推理
    • track: 跟踪
    • export: 导出
  • ARGS:
    • model: 模型路径
    • source: 数据源
    • class: 类别
    • comf: 置信度
    • device: 设备
      • cpu: 使用 CPU
      • gpu: 使用 GPU
      • mps: 使用 MPS [[macOS]]
      • cuda: 使用 CUDA

使用

命令行

推理任务

Terminal window
# 检测
yolo detect predict model=yolo11n.pt source='https://ultralytics.com/images/bus.jpg'
# 分割
yolo segment predict model=yolo11n-seg.pt source='https://ultralytics.com/images/bus.jpg'
# 分类
yolo classify predict model=yolo11n-cls.pt source='https://ultralytics.com/images/bus.jpg'
# 姿势
yolo pose predict model=yolo11n-pose.pt source='https://ultralytics.com/images/bus.jpg'
# OBB
yolo obb predict model=yolo11n-obb.pt source='https://ultralytics.com/images/boats.jpg'

导出模型

Terminal window
yolo export model=yolov8n-cls.pt format=onnx imgsz=224,128

训练

segment:

Terminal window
yolo segment train data=carparts-seg.yaml model=yolo11n-seg.pt epochs=100 imgsz=640

detect:

Terminal window
yolo detect train data=truck.yaml model=yolo11m.pt epochs=100 imgsz=640

查看模型类别

可以通过 [[Python]] 查看模型支持的类别:

Terminal window
python -c "from ultralytics import YOLO; model = YOLO('yolo11m.pt'); print('类别:', model.names)"