尧图网站建设 尧图网络
  • 首页
  • 关于我们
  • 服务项目
  • 案例展示
  • 建站流程
  • 资讯中心
  • 联系我们
首页/资讯中心/详情

从零开始构建yolov8-seg模型

从零开始构建yolov8-seg模型
📅 发布时间:2026/6/30 22:13:00

一、前言

本文为初学者提供思路,同时为自己后续能够回头翻阅,本文方法通用于其他yolo模型。

二、软件准备

这里用到是是Pycharm和Miniforge

PyCharm 是 Python 开发最常用的集成开发环境(IDE)。

Miniforge 是一个轻量级的 Python 环境和包管理器,可以看作是 Anaconda 的开源免费替代品。

自行查阅资料下载最新版本的软件。

下载后的Miniforge可能打不开,需要添加环境变量path

添加环境变量
D:\miniforge
D:\miniforge\Scripts
D:\miniforge\Library

具体的位置再进行对应的更改。

三、Miniforge 包管理

在这里需要在Miniforge中创建2个环境,当然放同一个也可以,但为了方便管理和之后的使用还是分为2个包。一个是labelme,这是为了标注数据的。一个是存放yolov8-seg运行所依赖的包

这里给到指令为

创建labelme环境 conda create -n labelme python=3.6 激活环境 conda activate labelme 安装labelme所需要的依赖环境 conda install pyqt conda install pillow 安装labelme pip install labelme==3.16.2

这样后创建了一个labelme的环境,之后打开labelme都需要在这个环境里,应该所需要的包都在这个环境。

之后就能打开标注软件,怎么进行数据的标注可以自行查询。

第二步yolo环境的搭建

在labelme环境中转换到网上下载的yolo实例代码带setup.py的目录下。

创建:conda create -n yolov8 python=3.8 查看:conda env list conda activate yolov8 切换阿里云 pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/ 安装依赖包 pip install -e . 安装 CMake conda install cmake -y 安装 ONNX 相关包 pip install onnx onnxruntime onnxsim

至此这个yolov8环境拥有了模型训练所需的所有依赖包。

四、模型训练

用Pycharm打开下载的yolo实例代码,同时将pycharm的环境转换为yolov8的环境

自此你就可以在pycharm中通过自己标注的数据集进行模型训练了。

这里的yolov8-seg.yaml为模型运行工程中进行的各类操作,后续可以对其进行修改,使其具备独特性。yolov8n-seg.pt是初始模型作用是使训练在一定的认知基础上进行训练,提高效率。coco128-seg.yaml就是我们的训练集了,大致内容可以是这样。

里面包括的训练集、验证集、测试集的存放位置以及类型。

运行后代码会在yolov8-seg\ultralytics\runs\segment目录下生成它的.pt文件这是pytroch的文件需要将其转换为onnx文件。可直接通过代码实现。

from ultralytics import YOLO # 加载分割模型 model = YOLO('D:/python/yolov8-seg/.venv/yolov8-seg/ultralytics/runs/segment/train8/weights/best.pt') # 分割模型 # 转换为ONNX model.export( format='onnx', imgsz=320, opset=18, simplify=True, dynamic=False # 固定输入尺寸 )

自此模型已经训练完成。

相关新闻

  • 容器化——让应用“拎包入住“
  • AI 编程框架全景比较 - 使用场景、优势与选型指南
  • 【我是如何在一个电商平台上发现一个高危IDOR漏洞的】

最新新闻

  • MySQL实战指南:从SQL语法到索引优化与生产环境调优
  • 计算机毕业设计之基于SSM的校园共享单车管理系统设计与实现
  • RAG 真正让人头疼的地方,从来不是“搭不起来”
  • UE 移动端 CPU、GPU、内存问题怎么归因:一套性能分析方法
  • 抖音无水印下载技术解析:从录屏到原生文件获取的革命
  • AI 创业融资策略:从技术壁垒到资本叙事的结构化拆解

日新闻

  • 【计算机毕业设计案例】基于 Spring Boot+Vue 的电影售票系统设计与实现 前后端分离架构下影院在线购票管理平台(程序+文档+讲解+定制)
  • 到底 TMD 用哪个: npm, pnpm, Yarn, Bun, Deno? 傻瓜, 当然用 npm 啦
  • Google限制Meta使用Gemini模型 凸显AI授权竞争白热化

周新闻

  • Windows字体自定义终极方案:No!! MeiryoUI完全指南
  • Deepin Boot Maker:告别命令行,3分钟制作Linux启动盘的智能解决方案
  • Plain Craft Launcher 2:重新定义你的Minecraft游戏体验

月新闻

  • 【总结】入门篇:50句话让你记住架构核心概念
  • WeChatMsg技术方案解析:实现Mac微信数据自主管理的完整解决方案
  • WeChatMsg:革新性微信数据备份方案,打造你的专属数字记忆库

关于尧图

  • 公司简介
  • 团队介绍
  • 企业文化
  • 荣誉资质

服务项目

  • 定制开发
  • 电商建站
  • UI 设计
  • 运维服务

快速链接

  • 案例展示
  • 建站流程
  • 常见问题
  • 资讯中心

联系方式

  • 📍北京市朝阳区互联网产业园 A 座 10 层
  • 📞400-888-8888
  • ✉️contact@rkmt.cn
  • 🕐周一至周日 9:00-21:00

© 2024 北京尧图网络科技有限公司 版权所有 | 京 ICP 备 XXXXXXXX 号