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

结果

相关推荐
叶凡要飞1 小时前
RTX5060Ti安装双系统ubuntu22.04各种踩坑点(黑屏,引导区修复、装驱动、server版本安装)
人工智能·python·yolo·ubuntu·机器学习·操作系统
FL16238631298 小时前
无人机视角河道多目标垃圾检测数据集VOC+YOLO格式1736张6类别
yolo·无人机
XIAO·宝9 小时前
深度学习------YOLOV1和YOLOV2
人工智能·深度学习·yolo
禾昂.11 小时前
从 YOLO V1 到 V2:目标检测领域的一次关键技术迭代
yolo·目标检测·目标跟踪
Python图像识别21 小时前
71_基于深度学习的布料瑕疵检测识别系统(yolo11、yolov8、yolov5+UI界面+Python项目源码+模型+标注好的数据集)
python·深度学习·yolo
OAFD.1 天前
YOLOv3 详解:核心改进、网络架构与目标检测实践
网络·yolo·目标检测
一百天成为python专家1 天前
python爬虫入门(小白五分钟从入门到精通)
开发语言·爬虫·python·opencv·yolo·计算机视觉·正则表达式
JANGHIGH1 天前
YOLO系列——OpenCV DNN模块在YOLOv11检测物体时输出的边界框坐标问题
opencv·yolo·dnn
宁若风1 天前
如何将yolov5模型部署到RK3588开发板上
yolo
尤超宇1 天前
YOLOv3 目标检测算法核心技术
算法·yolo·目标检测