YOLOv8:Roboflow公开数据集训练模型

2024-03-24 23:04

本文主要是介绍YOLOv8:Roboflow公开数据集训练模型,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

 Roboflow公开数据集

Roboflow是一个提供计算机视觉数据集管理和处理工具的平台。虽然Roboflow本身并不创建或策划公开数据集,但它提供了一系列功能,帮助用户组织、预处理、增强和导出计算机视觉数据集。

官方网站:https://universe.roboflow.com/

然而,有几个常用的公开数据集可供计算机视觉使用,可以在Roboflow或其他计算机视觉平台上访问和使用这些数据集。以下是一些常用的计算机视觉公开数据集:

  1. COCO(Common Objects in Context):COCO是一个大规模数据集,其中包含带有对象注释的图像,可用于对象检测、分割和字幕生成等任务。

  2. ImageNet:ImageNet是一个包含数百万标记图像的数据集,涵盖了数千个类别。它被广泛用于图像分类和深度学习研究。

  3. Open Images:Open Images是一个包含数百万图像及其对象检测、分割和视觉关系注释的数据集。

  4. Pascal VOC:Pascal VOC数据集是一个包含图像及其对象检测、分割和分类注释的集合。它常用于计算机视觉算法的基准测试。

  5. Cityscapes:Cityscapes是一个专注于城市场景的数据集,包含高质量图像以及像素级别的语义分割和实例分割注释。

  6. LFW(Labeled Faces in the Wild):LFW是一个包含从网络收集的人脸图像的数据集。它常用于人脸识别任务。

部署安装YOLOv8环境

Github官方网址:https://github.com/ultralytics/ultralytics

安装和部署YOLOv8:http://t.csdnimg.cn/iGwXY

下载Roboflow公开数据集

打开官方网站:https://universe.roboflow.com/

 选择自己需要的数据集:

点击左旁工具栏的Dataset:

 由于我们需要的是在YOLOv8下训练的数据集,故点击YOLOv8

 

 在窗口中选择download zip to computer(下载压缩包至电脑)

即可下载完成

训练数据集

打开pycharm,在安装好的YOLOv8项目下新建datasets文件夹

 将下载好的Roboflow公开数据集(我下载的animals数据集)解压到datasets文件夹中,目录格式如下:

 注:coco128是另外一个数据集。

此时打开animals数据集文件夹中的data.yaml文件:

train: ../train/images
val: ../valid/images
test: ../test/imagesnc: 10
names: ['cat', 'chicken', 'cow', 'dog', 'fox', 'goat', 'horse', 'person', 'racoon', 'skunk']roboflow:workspace: roboflow-100project: animals-ij5d2version: 2license: CC BY 4.0url: https://universe.roboflow.com/roboflow-100/animals-ij5d2/dataset/2

此数据集共分为10类,训练、测试的地址都有给出。

在确保文件目录格式正确,存在yaml文件之后,打开pycharm终端控制器:

一定要进入所建的YOLOv8环境中。

模型训练在终端运行:

yolo detect train data=datasets/animals/data.yaml model=yolov8n.yaml pretrained=yolov8n.pt epochs=100 batch=4 lr0=0.01 resume=True

注意:data=后要填写数据集文件夹中的yaml文件的绝对地址,相对地址可能会报错。

当然在训练的代码中会有许多参数,以上所填的是常用的训练参数,完整的训练参数如下:

 

 

 

 可参考YOLOv8文档:https://docs.ultralytics.com/modes/train/

 按回车即可开始训练数据集,系统将自动将训练好的模型保存至runs/detect/train文件夹下:

 其中weights文件中为,最后一次训练的模型last.pt以及效果最好的一次模型best.pt

以及数据集的标签图等:

 验证数据集

 在终端命令行输入:

yolo detect val data=E:\python_project\ultralytics-main\datasets\animals\data.yaml model=runs/detect/train/weights/best.pt batch=4

产生输出:

 其验证结果将保存至runs\detect\val文件夹下:

 可以看出训练的模型产生了较好的预测结果。

模型导出

使用下面的命令就可以导出模型:

yolo task=detect mode=export model=ultralytics/yolo/v8/detect/runs/detect/train/weights/best.pt 

参考

2023最新-用yolov8训练自己的数据集

http://t.csdnimg.cn/q6Gbb

YOLOv8教程系列:一、使用自定义数据集训练YOLOv8模型

http://t.csdnimg.cn/H5et2

这篇关于YOLOv8:Roboflow公开数据集训练模型的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/843070

相关文章

使用Java解析JSON数据并提取特定字段的实现步骤(以提取mailNo为例)

《使用Java解析JSON数据并提取特定字段的实现步骤(以提取mailNo为例)》在现代软件开发中,处理JSON数据是一项非常常见的任务,无论是从API接口获取数据,还是将数据存储为JSON格式,解析... 目录1. 背景介绍1.1 jsON简介1.2 实际案例2. 准备工作2.1 环境搭建2.1.1 添加

MySQL中删除重复数据SQL的三种写法

《MySQL中删除重复数据SQL的三种写法》:本文主要介绍MySQL中删除重复数据SQL的三种写法,文中通过代码示例讲解的非常详细,对大家的学习或工作有一定的帮助,需要的朋友可以参考下... 目录方法一:使用 left join + 子查询删除重复数据(推荐)方法二:创建临时表(需分多步执行,逻辑清晰,但会

Java实现任务管理器性能网络监控数据的方法详解

《Java实现任务管理器性能网络监控数据的方法详解》在现代操作系统中,任务管理器是一个非常重要的工具,用于监控和管理计算机的运行状态,包括CPU使用率、内存占用等,对于开发者和系统管理员来说,了解这些... 目录引言一、背景知识二、准备工作1. Maven依赖2. Gradle依赖三、代码实现四、代码详解五

详谈redis跟数据库的数据同步问题

《详谈redis跟数据库的数据同步问题》文章讨论了在Redis和数据库数据一致性问题上的解决方案,主要比较了先更新Redis缓存再更新数据库和先更新数据库再更新Redis缓存两种方案,文章指出,删除R... 目录一、Redis 数据库数据一致性的解决方案1.1、更新Redis缓存、删除Redis缓存的区别二

Redis事务与数据持久化方式

《Redis事务与数据持久化方式》该文档主要介绍了Redis事务和持久化机制,事务通过将多个命令打包执行,而持久化则通过快照(RDB)和追加式文件(AOF)两种方式将内存数据保存到磁盘,以防止数据丢失... 目录一、Redis 事务1.1 事务本质1.2 数据库事务与redis事务1.2.1 数据库事务1.

Oracle Expdp按条件导出指定表数据的方法实例

《OracleExpdp按条件导出指定表数据的方法实例》:本文主要介绍Oracle的expdp数据泵方式导出特定机构和时间范围的数据,并通过parfile文件进行条件限制和配置,文中通过代码介绍... 目录1.场景描述 2.方案分析3.实验验证 3.1 parfile文件3.2 expdp命令导出4.总结

更改docker默认数据目录的方法步骤

《更改docker默认数据目录的方法步骤》本文主要介绍了更改docker默认数据目录的方法步骤,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一... 目录1.查看docker是否存在并停止该服务2.挂载镜像并安装rsync便于备份3.取消挂载备份和迁

不删数据还能合并磁盘? 让电脑C盘D盘合并并保留数据的技巧

《不删数据还能合并磁盘?让电脑C盘D盘合并并保留数据的技巧》在Windows操作系统中,合并C盘和D盘是一个相对复杂的任务,尤其是当你不希望删除其中的数据时,幸运的是,有几种方法可以实现这一目标且在... 在电脑生产时,制造商常为C盘分配较小的磁盘空间,以确保软件在运行过程中不会出现磁盘空间不足的问题。但在

Python基于火山引擎豆包大模型搭建QQ机器人详细教程(2024年最新)

《Python基于火山引擎豆包大模型搭建QQ机器人详细教程(2024年最新)》:本文主要介绍Python基于火山引擎豆包大模型搭建QQ机器人详细的相关资料,包括开通模型、配置APIKEY鉴权和SD... 目录豆包大模型概述开通模型付费安装 SDK 环境配置 API KEY 鉴权Ark 模型接口Prompt

Java如何接收并解析HL7协议数据

《Java如何接收并解析HL7协议数据》文章主要介绍了HL7协议及其在医疗行业中的应用,详细描述了如何配置环境、接收和解析数据,以及与前端进行交互的实现方法,文章还分享了使用7Edit工具进行调试的经... 目录一、前言二、正文1、环境配置2、数据接收:HL7Monitor3、数据解析:HL7Busines