news 2026/6/30 22:11:47

从零开始构建yolov8-seg模型

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
从零开始构建yolov8-seg模型

一、前言

本文为初学者提供思路,同时为自己后续能够回头翻阅,本文方法通用于其他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 # 固定输入尺寸 )

自此模型已经训练完成。

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/6/30 22:02:07

MySQL新手入门实战:从零搭建环境到掌握核心CRUD与查询优化

这类教程最值得先看的不是功能列表,而是能不能帮你把环境搭起来、把基础命令跑通,再一步步理解数据怎么存、怎么查、怎么改。很多人一上来就找“大全”或“精通”,结果连本地MySQL服务都启动不了,或者建个表都报错。我更建议把学习…

作者头像 李华
网站建设 2026/6/30 21:58:35

C++课后习题训练记录Day144

1.练习项目 :问题描述小明是蓝桥王国的王子,今天是他登基之日。在即将成为国王之前,老国王给他出了道题,他想要考验小明是否有能力管理国家。题目的内容如下:蓝桥王国一共有 N 个建筑和 M 条单向道路,每条…

作者头像 李华
网站建设 2026/6/30 21:57:52

MySQL课程SQL知识点总结|语句规范、实战案例、易错点与学习复盘

MySQL课程SQL知识点总结|语句规范、实战案例、易错点与学习复盘 一、前言 本学期《MySQL数据库技术》课程系统学习了数据库设计、数据库定义、数据操作、数据查询、视图、索引分区、数据库编程、事务控制、权限管理、数据备份恢复全套核心知识点。课程搭配大量实操作…

作者头像 李华
网站建设 2026/6/30 21:53:02

网盘直链下载助手终极指南:3分钟实现全速下载自由

网盘直链下载助手终极指南:3分钟实现全速下载自由 【免费下载链接】Online-disk-direct-link-download-assistant 一个基于 JavaScript 的网盘文件下载地址获取工具。基于【网盘直链下载助手】修改 ,支持 百度网盘 / 阿里云盘 / 中国移动云盘 / 天翼云盘…

作者头像 李华
网站建设 2026/6/30 21:52:08

MySQL 临时表性能优化

MySQL临时表性能优化实战指南 在数据库应用中,临时表是处理复杂查询和中间结果的常用工具,但不当使用可能导致性能瓶颈。本文将从实际场景出发,解析如何通过优化临时表提升MySQL执行效率,尤其适合高并发或大数据量场景的开发人员…

作者头像 李华