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

结果

相关推荐
@Mr_LiuYang16 小时前
用自己的数据训练yolov11目标检测
yolo·目标检测·实例分割·yolov11
子正19 小时前
yolo11s rknn无法detect的bugfix - step by step
yolo·机器学习·ai
子正4 天前
pt->onnx->rknn(量化) step by step & FAQ
python·深度学习·yolo·ai
欣然~4 天前
使用 YOLOv8 模型分析摄像头的图像
yolo
Hello Mr.Z4 天前
在香橙派5 NPU上使用Yolov5
yolo·npu·香橙派5
goomind6 天前
YOLOv11实战海洋动物图像识别
pytorch·深度学习·yolo·计算机视觉·cnn·海洋动物
前网易架构师-高司机7 天前
木材表面缺陷检测数据集,支持YOLO+COCO JSON+PASICAL VOC XML+DARKNET格式标注信息,平均正确识别率95.0%
yolo·数据集·木材缺陷识别·木材表面缺陷
goomind8 天前
深度卷积神经网络实战海洋动物图像识别
深度学习·神经网络·yolo·计算机视觉·cnn·pyqt5·海洋动物识别
糖炒狗子8 天前
基于YoloV11和驱动级鼠标模拟实现Ai自瞄
人工智能·yolo·计算机外设
飞瀑8 天前
计算机视觉核心任务
yolo