当前位置: 首页 > news >正文

YOLOv8模型训练过程

一,conda环境的创建就略过了

        先进行库工具安装

        pip install ultralytics -i https://pypi.tuna.tsinghua.edu.cn/simple
二,在github上下载好模型,先预测一把
        yolo predict model=yolov8n.pt source=ultralytics/assets/bus.jpg

        如果无法自动下载yolov8n.pt可以先在别的地方下载好,再拖到项目下,在model=下给出文件地址即可

三,准备自己数据集的格式

data    (从这里可以看出,在训练时,去同级labels中去寻找对应的标注txt文件的)
——train
    ——images(一个个的单张图片)
    ——labels  (名字相同的txt文件)
——val
    ——imags
    ——labels

四,yaml(这个也可以放绝对路径,绝对路径更安全一些)
        train: /data/train
        val: /data/val
        test: /data/test

# number of classes
nc: 1

五,单卡训练指令

yolo task=detect mode=train model=yolov8n.pt data=E:\yolov8\dataset_2.yaml batch=15 epochs=10 imgsz=640 workers=0 device=cpu
六,针对图片进行预测 ——source可以是单张图片,也可以是一个文件夹
yolo task=detect mode=predict model=runs/detect/train/weights/best.pt source=data2/test device=cpu
七,onnx的导出
需要安装 onnxslim  onnxruntime
pip install  onnxslim  onnxruntime -i https://mirrors.aliyun.com/pypi/simple/
导出onnx的代码
yolo task=detect mode=export model=runs/detect/train/weights/best.pt format=onnx

八,输出结构介绍

通过NEtron看一下输出结构 输出1*5*8400
1*(center_x,center_y, w, h+n*class(config))*8400  这里只有一个类别,所以是4+1=5
640*640的图片下采样8倍,16倍,32倍的预测
80*80+40*40+20*20=8400
http://www.xdnf.cn/news/246745.html

相关文章:

  • Python与MySQL高效集成指南:从基础到高级实践
  • Hibernate与MybatisPlus的混用问题(Invalid bound statement (not found))
  • (C题|社交媒体平台用户分析问题)2025年第二十二届五一数学建模竞赛(五一杯/五一赛)解题思路|完整代码论文集合
  • 恒流源电路
  • RAG工程-基于LangChain 实现 Advanced RAG(预检索-查询优化)(下)
  • 前端HTML基础知识
  • 【大模型面试每日一题】Day 5:GQA vs MHA效率对比
  • CV中常用Backbone-2:ConvNeXt模型详解
  • 微软推出数款Phi 4“开放式”人工智能模型
  • VB.net序列化和反序列化的使用方法和实用场景
  • NUS:多模态多视角理解评估
  • 攻防世界 - Misc - Level 6 | Wireshark
  • jupyterlab建议安装的两个插件
  • LeetCode:DP-回文串问题
  • 如何测试登录模块?全面测试思路解析
  • Python爬虫(14)Python爬虫数据存储新范式:云原生NoSQL服务实战与运维成本革命
  • Socket通信
  • Beetle-RP2350 扩展板设计
  • 力扣——23合并升序链表
  • 【ESP32】st7735s + LVGL使用-------图片显示
  • python多线程输入字符和写入文件
  • 关系型数据库设计指南
  • 2025五一杯数学建模竞赛选题建议+初步分析
  • terraform实现本地加密与解密
  • sftp连接报错Received message too long 168449893
  • 大鱼吃小鱼开源
  • leetcode 977. Squares of a Sorted Array
  • 【免费】1992-2021年各省GDP数据/各省地区生产总值数据
  • GoogleTest:简单示例及ASSERT/EXPECT说明
  • [FPGA 官方 IP] Binary Counter