YOLO 自己训练一个模型

一、准备数据集

我的版本是yolov8 8.11

这个目录结构很重要

cpp 复制代码
ultralytics-main  
              |  datasets
                         |coco
                              |train
                              |val

二、训练

编写yaml 文件

cpp 复制代码
# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
path: "D:\\work\\pycharmworkplace\\ultralytics-main\\datasets\\coco"  # dataset root dir
train: "D:\\work\\pycharmworkplace\\ultralytics-main\\datasets\\coco\\train"
val: "D:\\work\\pycharmworkplace\\ultralytics-main\\datasets\\coco\\val"
#test:  # test images (optional)

# Classes (80 COCO classes)
names:
  0: fire

编写python 文件

cpp 复制代码
from ultralytics import YOLO

# Load a model
model = YOLO('.\\ultralytics-main\\yolov8n.pt')  # load a pretrained model (recommended for training)

# Train the model
results = model.train(data='.\\ultralytics-main\\datasets\\coco\\coco.yaml', epochs=10, imgsz=640)

# 检测命令
# yolo predict model=best.pt  source=ultralytics\assets\1_5.jpg

三、验证

自动训练

用python 文件启动

我只总共2600 张图,训练集是1600 张 ,跑10次大概一个多小时跑完

在dataset 的那个文件夹的coco 文件夹下生成了一个runs 的文件夹,里面就有模型和结果

结果看不懂,后面再说,现在找模型

训练的模型在这个目录下

测试:

测试命令

cpp 复制代码
yolo predict model=.datasets\\coco\\runs\\detect\\train2\\weights\\best.pt   source=ultralytics\assets
\1_54.jpg

结果

相关推荐
学技术的大胜嗷2 小时前
离线迁移 Conda 环境到 Windows 服务器:用 conda-pack 摆脱硬路径限制
人工智能·深度学习·yolo·目标检测·机器学习
一花·一叶1 天前
基于昇腾310B4的YOLOv8目标检测推理
yolo·目标检测·边缘计算
昵称是6硬币1 天前
YOLOv11: AN OVERVIEW OF THE KEY ARCHITECTURAL ENHANCEMENTS目标检测论文精读(逐段解析)
图像处理·人工智能·深度学习·yolo·目标检测·计算机视觉
OICQQ676580081 天前
创建一个基于YOLOv8+PyQt界面的驾驶员疲劳驾驶检测系统 实现对驾驶员疲劳状态的打哈欠检测,头部下垂 疲劳眼睛检测识别
yolo·pyqt·疲劳驾驶·检测识别·驾驶员检测·打哈欠检测·眼睛疲劳
king of code porter10 天前
目标检测之YOLOv5到YOLOv11——从架构设计和损失函数的变化分析
人工智能·yolo·目标检测
model200511 天前
yolov11转ncnn
yolo·ncnn
YueiL11 天前
ROS 2 中 Astra Pro 相机与 YOLOv5 检测功能编译启动全记录
yolo·ros2
来两个炸鸡腿12 天前
【Datawhale组队学习202506】YOLO-Master task03 IOU总结
python·学习·yolo
凌佚12 天前
rknn优化教程(三)
c++·yolo·目标检测
毕设做完了吗?12 天前
基于YOLO的智能车辆检测与记录系统
yolo