docker快速使用yolov11

此文章为在h200上dood模式下使用yolov11训练模型【其他机器大同小异】

拉取镜像

powershell 复制代码
docker pull ultralytics/ultralytics:latest

启动容器

powershell 复制代码
docker run -d --rm --name container-name --gpus "device=9" --ipc=host -p 8456:8888  -it ultralytics/ultralytics:latest

设定训练结果目录

如果不设置,yolo训练完会放默认目录下,训练日志最后会提示

powershell 复制代码
yolo settings runs_dir=/workspace/runs

创建dataset目录

· dataset目录格式如下,创建好后,把训练集和验证集数据拷贝进去

· 或者也可在windows弄好,再拷贝进容器后解压(我的服务器不支持挂载)

· 如果支持挂载目录直接在windows下弄好dataset即可

powershell 复制代码
dataset/
├── images/
│   ├── train/
│   │   ├── img_001.jpg
│   │   └── ...
│   └── val/
│       ├── img_201.jpg
│       └── ...
├── labels/
│   ├── train/
│   │   ├── img_001.txt
│   │   └── ...
│   └── val/
│       ├── img_201.txt
│       └── ...
└── dataset.yaml

dataset.yaml内容如下,请按实际情况修改

powershell 复制代码
path: /workspace/dataset
train: images/train
val: images/val

nc: 5
names: [block, 1s, 1w, 1n, 1e]

开始训练

· 使用yolo11s.pt作为初始模型,训练100轮,图像尺寸1280,每个批次处理图像个数

· 其他参数诸如lr0学习率、patience早停耐心值等请按需添加

· h200网络没有问题的话,会自动下载yolo11s.pt,无需关心。如果网络有问题就得自己想办法下载了。

powershell 复制代码
yolo train data=/workspace/dataset/dataset.yaml model=yolo11s.pt epochs=100 imgsz=1280 batch=4

训练完成

训练完成后去刚刚设定的结果目录下分析训练结果。如果没有设定,则根据日志去默认目录下查看训练结果。

相关推荐
I'm a winner14 小时前
基于YOLO算法的医疗应用专题:第一章 计算机视觉与深度学习概述
算法·yolo·计算机视觉
java1234_小锋1 天前
[免费]基于Python的YOLO深度学习垃圾分类目标检测系统【论文+源码】
python·深度学习·yolo·垃圾分类·垃圾分类检测
AI棒棒牛1 天前
论文精读系列:Retinanet——目标检测领域中的SCI对比实验算法介绍!可一键跑通的对比实验,极大节省小伙伴的时间!!!
yolo·目标检测·计算机视觉·对比实验·1024程序员节·创新·rtdter
遇雪长安1 天前
深度学习YOLO实战:4、模型的三要素:任务、类别与规模
人工智能·深度学习·yolo
侯喵喵1 天前
Jetson orin agx配置ultralytics 使用docker或conda
yolo·docker·1024程序员节·ultralytics
tainshuai2 天前
YOLOv4 实战指南:单 GPU 训练的目标检测利器
yolo·目标检测·机器学习
飞翔的佩奇2 天前
【完整源码+数据集+部署教程】【运动的&足球】足球场地区域图像分割系统源码&数据集全套:改进yolo11-RFAConv
前端·python·yolo·计算机视觉·数据集·yolo11·足球场地区域图像分割系统
夏天是冰红茶2 天前
恶劣天气目标检测IA-YOLO
yolo·目标检测·目标跟踪
MavenTalk2 天前
如何根据不同的场景选择YOLO相应的基座模型
python·yolo·yolo11n·yolo11m·yolo11s·yolo11x
LiJieNiub2 天前
深入解析 YOLOv5 datasets.py:数据加载与增强的核心逻辑
yolo