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

结果

相关推荐
Suyuoa1 小时前
附录2-pytorch yolov5目标检测
python·深度学习·yolo
红色的山茶花20 小时前
YOLOv8-ultralytics-8.2.103部分代码阅读笔记-block.py
笔记·深度学习·yolo
unix2linux1 天前
YOLO v5 Series - Image & Video Storage ( Openresty + Lua)
yolo·lua·openresty
菠菠萝宝1 天前
【YOLOv8】安卓端部署-1-项目介绍
android·java·c++·yolo·目标检测·目标跟踪·kotlin
ZZZZ_Y_1 天前
YOLOv5指定标签框背景颜色和标签字
yolo
红色的山茶花2 天前
YOLOv8-ultralytics-8.2.103部分代码阅读笔记-conv.py
笔记·yolo
Eric.Lee20212 天前
数据集-目标检测系列- 花卉 鸡蛋花 检测数据集 frangipani >> DataBall
人工智能·python·yolo·目标检测·计算机视觉·鸡蛋花检查
阿_旭2 天前
【模型级联】YOLO-World与SAM2通过文本实现指定目标的零样本分割
yolo·yolo-world·sam2
CSBLOG2 天前
OpenCV、YOLO、VOC、COCO之间的关系和区别
人工智能·opencv·yolo
2zcode2 天前
基于YOLOv8深度学习的医学影像骨折检测诊断系统研究与实现(PyQt5界面+数据集+训练代码)
人工智能·深度学习·yolo