Python Package

ultralytics

2024-02-01

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

安装

Terminal window
pip install ultralytics

使用

命令行

语法:

Terminal window
yolo TASK MODE ARGS

推理任务

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

训练

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