YOLO26社区资源汇总:GitHub、论坛、教程链接集

本镜像基于 YOLO26 官方代码库 构建,预装了完整的深度学习开发环境,集成了训练、推理及评估所需的所有依赖,开箱即用。

1. 镜像环境说明

  • 核心框架: pytorch == 1.10.0
  • CUDA版本: 12.1
  • Python版本: 3.9.5
  • 主要依赖: torchvision==0.11.0, torchaudio==0.10.0, cudatoolkit=11.3, numpy, opencv-python, pandas, matplotlib, tqdm, seaborn 等。

该镜像专为 YOLO26 模型的快速部署与实验验证设计,避免了繁琐的环境配置过程。所有常用计算机视觉和数据处理库均已预装,用户可直接进入模型训练与推理阶段,极大提升研发效率。


2. 快速上手

启动界面示意图

镜像启动后默认进入 Jupyter 或终端交互界面(视平台而定),如下图所示:

工作台界面

2.1 激活环境与切换工作目录

在使用前,请先激活 Conda 环境以确保所有依赖正确加载:

conda activate yolo

环境激活命令执行

由于系统盘空间有限且不可持久化保存,建议将代码复制到数据盘进行修改与长期存储。执行以下命令完成复制操作:

cp -r /root/ultralytics-8.4.2 /root/workspace/

代码复制操作

随后进入目标目录开始后续操作:

cd /root/workspace/ultralytics-8.4.2

进入代码目录 目录结构展示


2.2 模型推理

YOLO26 支持多种任务类型(如目标检测、姿态估计等),通过调用 ultralytics 库即可实现一键推理。

首先修改 detect.py 文件内容如下:

# -*- coding: utf-8 -*-
"""
@Auth :落花不写码
@File :detect.py
@IDE :PyCharm
@Motto :学习新思想,争做新青年
"""

from ultralytics import YOLO

if __name__ == '__main__':

    # Load a model
    model = YOLO(model=r'yolo26n-pose.pt')
    model.predict(source=r'./ultralytics/assets/zidane.jpg',
                  save=True,
                  show=False,
                  )
推理参数详解
  • model 参数:指定模型权重文件路径,支持 .pt 格式的预训练权重。
  • source 参数:输入源路径,可以是本地图片、视频文件路径,或摄像头编号(如 0 表示默认摄像头)。
  • save 参数:设为 True 时,自动保存推理结果图像至 runs/detect/predict/ 目录。
  • show 参数:是否实时显示推理窗口,服务器环境下建议设为 False 以避免图形界面报错。

运行以下命令启动推理:

python detect.py

推理执行过程 推理结果输出

推理结果将在终端打印详细信息,并根据设置保存至指定路径,便于后续分析与可视化。


2.3 模型训练

要使用自定义数据集进行训练,需准备符合 YOLO 格式的数据标注文件,并更新配置。

数据集配置

上传你的数据集并组织成如下结构:

dataset/
├── images/
│   ├── train/
│   └── val/
├── labels/
│   ├── train/
│   └── val/
└── data.yaml

然后编辑 data.yaml 文件,示例如下:

train: ./dataset/images/train
val: ./dataset/images/val

nc: 80
names: ['person', 'bicycle', 'car', ...]

data.yaml 编辑界面

参数说明:

  • train: 训练集图像路径
  • val: 验证集图像路径
  • nc: 类别数量
  • names: 类别名称列表
训练脚本配置

创建或修改 train.py 脚本:

# -*- coding: utf-8 -*-
"""
@Auth :落花不写码
@File :train.py
@IDE :PyCharm
@Motto :学习新思想,争做新青年
"""
import warnings
warnings.filterwarnings('ignore')
from ultralytics import YOLO

if __name__ == '__main__':
    model = YOLO(model='/root/workspace/ultralytics-8.4.2/ultralytics/cfg/models/26/yolo26.yaml')
    model.load('yolo26n.pt')  # 加载预训练权重(可选)
    model.train(data=r'data.yaml',
                imgsz=640,
                epochs=200,
                batch=128,
                workers=8,
                device='0',
                optimizer='SGD',
                close_mosaic=10,
                resume=False,
                project='runs/train',
                name='exp',
                single_cls=False,
                cache=False,
                )

关键参数解释:

  • imgsz: 输入图像尺寸,默认 640×640
  • epochs: 总训练轮数
  • batch: 批次大小,依据显存调整
  • device: 使用 GPU 设备编号
  • close_mosaic: 在最后若干 epoch 关闭 Mosaic 增强,提升收敛稳定性
  • resume: 是否从中断处恢复训练

执行训练命令:

python train.py

训练日志输出

训练过程中会自动生成日志、权重和可视化图表,保存路径由 projectname 参数决定。


2.4 下载训练结果

训练完成后,模型权重通常保存在 runs/train/exp/weights/ 目录下,包含 best.ptlast.pt 两个主要文件。

可通过 SFTP 工具(如 Xftp)将远程服务器上的结果下载至本地:

  1. 打开 Xftp 并连接当前实例;
  2. 在右侧(远程端)找到目标文件夹;
  3. 将其拖拽至左侧(本地端)完成下载;
  4. 单个文件也可双击直接下载。

Xftp 文件传输界面

建议对大文件先压缩再传输,例如使用 tar -czf exp.tar.gz runs/train/exp/ 减少网络耗时。


3. 已包含权重文件

镜像内已预下载常用 YOLO26 系列权重文件,存放于项目根目录,包括但不限于:

  • yolo26n.pt
  • yolo26s.pt
  • yolo26m.pt
  • yolo26l.pt
  • yolo26x.pt
  • yolo26n-pose.pt

预置权重文件列表

这些模型覆盖不同规模与应用场景(轻量级、高精度、姿态估计等),用户可根据实际需求选择加载,无需额外下载。


4. 常见问题

  • Q: 启动后无法导入 ultralytics?
    A: 请确认是否已执行 conda activate yolo 切换至正确环境。

  • Q: 训练时报错 “No module named ‘ultralytics’”?
    A: 检查当前路径是否位于 ultralytics-8.4.2 项目根目录,否则 Python 无法识别本地包。

  • Q: 如何更换 CUDA 版本?
    A: 当前镜像固定使用 CUDA 12.1 与 PyTorch 1.10.0 组合,不建议手动更改以免破坏依赖关系。

  • Q: 可否支持多卡训练?
    A: 支持。只需将 device 参数改为 '0,1,2,3' 形式即可启用多 GPU 分布式训练。

  • Q: 如何提高训练速度?
    A: 可尝试增大 batch 大小、开启 cache=True 缓存数据、关闭 show 和日志冗余输出。


5. 参考资料

  • 官方仓库: ultralytics/ultralytics
  • 文档说明: 详细用法请参考官方库中的 README.md 文件,涵盖 API 文档、训练技巧、导出部署等内容。
  • 社区支持:
    • GitHub Issues: https://github.com/ultralytics/ultralytics/issues
    • 官方 Discord: https://discord.gg/ultralytics
    • 中文技术交流群(非官方):可通过 CSDN、知乎等平台搜索加入活跃开发者社群。

获取更多AI镜像

想探索更多AI镜像和应用场景?访问 CSDN星图镜像广场,提供丰富的预置镜像,覆盖大模型推理、图像生成、视频生成、模型微调等多个领域,支持一键部署。

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐