当前位置: 首页 > 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

相关文章:

  • MCP:智能家居的“大脑”,如何引领未来居住革命
  • LangChain4j +DeepSeek大模型应用开发——7 项目实战 创建硅谷小鹿
  • Python Cookbook-6.14 实现状态设计模式
  • DeepSeek 赋能自然语言处理:从理论到实践的全方位解析
  • Qt指南针
  • 支持selenium的chrome driver更新到136.0.7103.49
  • 打包 Python 项目为 Windows 可执行文件:高效部署指南
  • 33、VS中提示“以下文件中的行尾不一致。是否将行尾标准化?“是什么意思?
  • 【C语言练习】014. 使用数组作为函数参数
  • Java关键字解析
  • Canvas特效实例:黑客帝国-字母矩阵(字母雨)
  • OpenAI最新发布的GPT-4.1系列模型,性能体验如何?
  • 设计模式简述(十四)组合模式
  • Unity SpriteMask(精灵遮罩)
  • Go 语言中一个功能强大且广泛使用的数据验证库github.com/go-playground/validator/v10
  • Js扩展DOM、BOM、AJAX、事件、定时器
  • 系统架构设计师:设计模式概述
  • 为 Unity 项目添加自定义 USB HID 设备支持 (适用于 PC 和 Android/VR)-任何手柄、无人机手柄、摇杆、方向盘
  • 补题( Convolution, 二维卷积求输出矩阵元素和最大值)
  • 华为云Astro大屏连接器创建操作实例:抽取物联网iotda影子设备数据的连接器创建
  • 戴上XR头盔,五一假期在上海也能体验“登陆月球”
  • 2025五一档首日电影票房破亿
  • 伊朗外长:伊美第四轮间接谈判将于5月3日举行
  • 山西太原一小区发生爆炸,造成1人遇难21人受伤2人失联
  • 制定出台民营经济促进法有何重大意义?全国人大常委会法工委回应
  • 孙磊已任中国常驻联合国副代表、特命全权大使