YOLO11新手实战:跟着步骤完成第一个目标检测项目
1. 项目介绍与环境准备
1.1 YOLO11简介
YOLO11是当前最先进的目标检测算法之一,以其快速、准确的特点在计算机视觉领域广受欢迎。这个镜像提供了完整的YOLO11运行环境,包含所有必要的依赖项,让你可以立即开始目标检测项目。
1.2 环境准备
首先确保你已经获取了YOLO11镜像并成功启动。镜像提供了两种主要使用方式:
- Jupyter Notebook:适合交互式开发和实验
- SSH连接:适合直接命令行操作
选择你熟悉的方式进入环境。如果你是初学者,建议从Jupyter Notebook开始。
2. 快速开始YOLO11项目
2.1 进入项目目录
无论使用哪种方式连接,第一步都是进入YOLO11项目目录:
cd ultralytics-8.3.9/这个目录包含了YOLO11的所有源代码和示例文件。
2.2 准备数据集
目标检测项目需要准备标注好的数据集。YOLO格式的数据集应包含:
- 图像文件(.jpg/.png等)
- 对应的标注文件(.txt),每行格式为:
class_id x_center y_center width height - 数据集描述文件(data.yaml),定义训练集、验证集路径和类别信息
示例data.yaml文件内容:
train: /path/to/train/images val: /path/to/val/images nc: 80 # 类别数量 names: ['person', 'bicycle', 'car', ...] # 类别名称列表3. 训练你的第一个YOLO11模型
3.1 训练脚本配置
YOLO11提供了简单的训练接口。创建一个train.py文件,内容如下:
from ultralytics import YOLO # 加载预训练模型 model = YOLO('yolov11n.pt') # 使用nano版本的小模型 # 开始训练 results = model.train( data='data.yaml', # 数据集配置文件 imgsz=640, # 输入图像尺寸 epochs=100, # 训练轮次 batch=16, # 批量大小 device='0', # 使用GPU 0 workers=8 # 数据加载线程数 )3.2 启动训练
运行训练脚本:
python train.py训练过程中,控制台会输出损失值和评估指标,模型权重会自动保存在runs/train/目录下。
4. 评估与使用训练好的模型
4.1 模型评估
训练完成后,可以使用验证集评估模型性能:
metrics = model.val() # 在验证集上评估 print(metrics.box.map) # 打印mAP指标4.2 使用模型进行预测
加载训练好的模型进行目标检测:
results = model.predict('test.jpg') # 对单张图片预测 results[0].show() # 显示检测结果 # 或者对视频进行检测 results = model.predict('video.mp4', save=True) # 保存结果视频5. 常见问题与解决方案
5.1 训练问题排查
- 显存不足:减小batch大小或图像尺寸
- 训练不收敛:检查学习率,尝试更小的值
- 类别不平衡:使用类别权重或数据增强
5.2 性能优化建议
- 使用更大的模型(如yolov11s/m/l/x)获得更高精度
- 增加训练数据量或使用数据增强
- 调整输入图像尺寸(更大的尺寸通常精度更高)
6. 项目总结与进阶学习
通过本教程,你已经完成了:
- YOLO11环境的搭建与配置
- 数据集的准备与格式转换
- 模型的训练与评估
- 使用训练好的模型进行预测
要进一步学习YOLO11,可以:
- 尝试不同的模型架构和超参数
- 在自己的数据集上微调模型
- 探索YOLO11的其他功能,如实例分割
获取更多AI镜像
想探索更多AI镜像和应用场景?访问 CSDN星图镜像广场,提供丰富的预置镜像,覆盖大模型推理、图像生成、视频生成、模型微调等多个领域,支持一键部署。