1. 为什么选择YOLOv8进行目标检测
YOLOv8作为Ultralytics公司推出的最新目标检测算法,在精度和速度之间取得了更好的平衡。相比前代YOLOv5,v8版本在保持实时性的同时,mAP指标提升了约15%。我在工业质检项目中实测发现,对于1280x720分辨率的图像,RTX 3060显卡上能达到85FPS的推理速度。
这个算法特别适合需要实时处理的场景,比如:
- 智能监控中的异常行为识别
- 自动驾驶中的障碍物检测
- 生产线上的缺陷品筛选
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 环境配置与基础使用
2.1 安装与验证
推荐使用Python 3.8+环境,通过pip安装最新版:
bash复制pip install ultralytics
验证安装是否成功:
python复制import ultralytics
print(ultralytics.__version__) # 应输出8.x.x
注意:如果遇到CUDA相关错误,建议先单独安装对应版本的torch:
bash复制pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118
2.2 第一个检测示例
加载预训练模型进行图片检测:
python复制from ultralytics import YOLO
model = YOLO('yolov8n.pt') # 加载nano版本
results = model('bus.jpg') # 推理
results[0].show() # 显示结果
模型尺寸选择建议:
| 模型类型 | 参数量 | 适用场景 |
|---|---|---|
| yolov8n | 3.2M | 移动端/嵌入式 |
| yolov8s | 11.4M | 平衡型 |
| yolov8m | 26.3M | 服务器部署 |
| yolov8l | 43.7M | 高精度需求 |
3. 自定义数据集训练
3.1 数据准备规范
建议使用YOLO格式的标注:
code复制class_id center_x center_y widt
