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

训练完成

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

相关推荐
西西弗Sisyphus5 小时前
YOLO 11 图像分类推理 Web 服务
yolo·分类·yolo 11
飞翔的佩奇1 天前
【完整源码+数据集+部署教程】鸡只与养殖场环境物品图像分割: yolov8-seg等50+全套改进创新点发刊_一键训练教程_Web前端展示
python·yolo·计算机视觉·数据集·yolov8·yolo11·鸡只与养殖场环境物品图像分割
夏雨不在低喃1 天前
YOLOv8目标检测融合RFLA提高小目标准确率
人工智能·yolo·目标检测
程序猿小D3 天前
【完整源码+数据集+部署教程】【智慧工地监控】建筑工地设备分割系统: yolov8-seg-efficientViT
python·yolo·计算机视觉·数据集·yolov8·yolo11·建筑工地设备分割系统
小雪狼3 天前
RV1126 RKNN环境搭建记录
rnn·yolo
nju_spy3 天前
计算机视觉 - 物体检测(二)单阶段:YOLO系列 + SSD
人工智能·yolo·目标检测·计算机视觉·ssd·r-cnn·端到端检测
码猩3 天前
YOLO通用无人机目标检测框架
人工智能·yolo·目标检测
Hcoco_me3 天前
YOLO入门教程(番外):计算机视觉数学、编程基础
人工智能·yolo·计算机视觉
IT古董5 天前
【第五章:计算机视觉-项目实战之目标检测实战】2.目标检测实战:中国交通标志检测-(4)YOLOv8训练与测试
yolo·目标检测·计算机视觉