Ultralytics YOLOv8
Ultralytics
构建环境
- GPU
docker pull ultralytics/ultralytics:latest
- CPU
docker pull ultralytics/ultralytics:latest-cpu
- Apple Silicon
docker pull ultralytics/ultralytics:latest-arm64
本地安装
pip install ultralytics
基于 COCO128 数据集的目标检测范例
运行容器
git clone https://github.com/ultralytics/ultralytics.git
docker run --runtime=nvidia -it --name ultralytics -v `pwd`/ultralytics:/usr/src/ultralytics ultralytics/ultralytics:latest
yolo 命令的使用参数
yolo TASK MODE ARGS
训练模型
yolo train data=coco128.yaml model=yolov8n.pt
训练可视化(Comet) pip install comet_ml export