ultralytics 是一个用于目标检测的 [[Python]] 库。
安装
pip install ultralytics
使用
命令行
语法:
yolo TASK MODE ARGS
推理任务
# 检测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'
# OBByolo obb predict model=yolo11n-obb.pt source='https://ultralytics.com/images/boats.jpg'
导出模型
yolo export model=yolov8n-cls.pt format=onnx imgsz=224,128
训练
yolo segment train data=carparts-seg.yaml model=yolo11n-seg.pt epochs=100 imgsz=640