1. YOLO26 环境配置与安装
1.1 创建 YOLO26 虚拟环境
在开始使用 YOLO26 之前,强烈建议创建一个独立的 Python 虚拟环境。虚拟环境可以将 YOLO26 所需的依赖包与其他项目的依赖包隔离开来,避免版本冲突问题。这里我们推荐使用 Miniconda 来管理虚拟环境,因为它比原生 Python 的 venv 更轻量且功能更强大。
首先确保你已经安装了 Miniconda。如果尚未安装,可以从 Miniconda 官网下载对应操作系统的安装包。安装完成后,打开终端(Windows 用户使用 Anaconda Prompt),执行以下命令创建虚拟环境:
bash复制conda create -n YOLO26 python=3.8
选择 Python 3.8 版本是因为它与 YOLO26 的兼容性最好,能避免很多潜在的依赖冲突。创建完成后,激活环境:
bash复制conda activate YOLO26
提示:每次开始工作前都需要先激活这个环境。你可以通过命令提示符前的 (YOLO26) 标识来判断当前是否处于正确的环境中。
1.2 配置 PyCharm 编辑器
对于 Python 开发,PyCharm 是最受欢迎的 IDE 之一。以下是详细的配置步骤:
- 打开 PyCharm,创建新项目或打开已有项目
- 进入解释器设置:
- 方法一:右下角状态栏点击当前解释器 → "解释器设置"
- 方法二:File → Settings → Project → Python Interpreter
- 添加 Conda 环境:
- 选择"使用现有环境"
- 类型选择"Conda"
- 指定 conda.exe 路径(通常在 Miniconda 安装目录的 Scripts 文件夹下)
- 选择我们刚创建的 YOLO26 环境
配置完成后,PyCharm 会自动识别环境中的所有已安装包,并提供智能代码补全功能。
2. 安装 Ultralytics YOLO26
2.1 Pip 安装 Ultralytics
最简单的安装方式是通过 pip 安装官方发布的稳定版本:
bash复制pip install -U ultralytics
这个命令会自动安装 YOLO26 及其所有依赖项,包括 PyTorch、OpenCV 等。安装完成后,可以通过以下命令验证是否安装成功:
bash复制yolo help
如果看到 YOLO 命令的帮助信息输出,说明安装成功。
2.2 克隆 GitHub 仓库(可选)
如果你想使用最新的开发版代码,可以克隆官方 GitHub 仓库:
bash复制git clone https://github.com/ultralytics/ultralytics
cd ultralytics
pip install -e .
这种"可编辑模式"安装允许你直接修改源代码并立即生效,适合想要贡献代码或自定义模型的研究人员。
2.3 GPU 加速配置
虽然 pip 安装会自动安装 PyTorch,但默认可能是 CPU 版本。要启用 GPU 加速,建议先安装 GPU 版 PyTorch:
bash复制pip install torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 --index-url https://download.pytorch.org/whl/cu121
然后安装 ultralytics:
bash复制pip install -U ultralytics
验证 GPU 是否可用:
python复制import torch
print(torch.cuda.is_available()) # 应该返回 True
print(torch.version.cuda) # 应该显示 12.1
3. YOLO26 快速入门
3.1 命令行界面(CLI)使用
YOLO26 提供了强大的命令行接口,基本语法为:
bash复制yolo TASK MODE ARGS
常用示例:
- 使用预训练模型检测图像:
bash复制yolo predict model=yolo26n.pt source='bus.jpg'
- 在 COCO 数据集上训练模型:
bash复制yolo detect train data=coco.yaml model=yolo26n.pt epochs=100 imgsz=640
- 验证模型性能:
bash复制yolo val model=yolo26n.pt data=coco.yaml
3.2 Python API 使用
除了命令行,YOLO26 也提供了 Python 接口:
python复制from ultralytics import YOLO
# 加载模型
model = YOLO('yolo26n.pt') # 或 'yolo26n.yaml' 从头训练
# 预测
results = model('bus.jpg')
# 训练
results = model.train(data='coco.yaml', epochs=100, imgsz=640)
4. YOLO26 预训练模型
YOLO26 提供了多种预训练模型,适用于不同任务和设备:
4.1 目标检测模型
| 模型 | 尺寸(pixels) | mAPval 50-95 | 速度(CPU ONNX) | 参数量(M) |
|---|---|---|---|---|
| YOLO26n | 640 | 40.9 | 38.9ms | 2.4 |
| YOLO26s | 640 | 48.6 | 87.2ms | 9.5 |
| YOLO26m | 640 | 53.1 | 220.0ms | 20.4 |
4.2 模型下载方式
- 自动下载:运行代码时如果本地没有模型,会自动从官网下载
- 手动下载:从 Ultralytics 的 GitHub Releases 页面下载对应模型
5. 高级配置
5.1 自定义训练配置
创建 custom.yaml 文件配置训练参数:
yaml复制# 训练设置
lr0: 0.01 # 初始学习率
lrf: 0.01 # 最终学习率 = lr0 * lrf
epochs: 100
batch: 16
# 数据增强
hsv_h: 0.015 # 色调增强
hsv_s: 0.7 # 饱和度增强
fliplr: 0.5 # 水平翻转概率
然后通过命令行指定配置文件:
bash复制yolo detect train data=custom.yaml model=yolo26n.pt
5.2 模型导出
YOLO26 支持导出为多种格式:
bash复制yolo export model=yolo26n.pt format=onnx # 导出为ONNX
yolo export model=yolo26n.pt format=tensorrt # 导出为TensorRT
6. 常见问题解决
-
CUDA out of memory:
- 减小 batch size
- 使用更小的模型(yolo26n → yolo26s)
- 尝试混合精度训练(amp=True)
-
安装冲突:
- 确保使用干净的虚拟环境
- 按顺序安装: PyTorch → TorchVision → Ultralytics
-
预测结果不理想:
- 调整 conf 参数(默认0.25)
- 尝试更大的模型
- 在自己的数据上微调模型
经验分享:在实际项目中,我们通常会先用小模型(yolo26n)快速验证思路,确认可行后再换大模型提升精度。同时,合理的数据增强能显著提升模型泛化能力。
通过以上步骤,你应该已经成功搭建了 YOLO26 开发环境,并能够开始你的计算机视觉项目了。YOLO26 的模块化设计让它可以轻松适应各种应用场景,从简单的物体检测到复杂的实例分割任务都能胜任。