当前位置: 代码网 > 科技>人工智能>数据分析 > YOLOv10最详细全面讲解2- 目标检测-环境搭建、训练自己的数据集

YOLOv10最详细全面讲解2- 目标检测-环境搭建、训练自己的数据集

2024年07月28日 数据分析 我要评论
本人基于继续往下进行,主要包括环境搭建、训练自己的数据集。

yolov10没想到出来的如此之快,作为一名yolo的爱好者,以yolov5和yolov8的经验,打算出一套从数据集装备->环境配置->训练->验证->目标追踪全系列教程。请大家多多点赞和收藏!!!
系列文章:
yolov10最详细全面讲解1- 目标检测-准备自己的数据集(yolov5,yolov8均适用)

本人基于yolov10最详细全面讲解1- 目标检测-准备自己的数据集(yolov5,yolov8均适用)继续往下进行,主要包括环境搭建、训练自己的数据集。

1.环境搭建

1.1官方下载源码

官网地址:yolov10 gitbub官网源码
利用魔法进入github官网之后点击下载源码压缩包(这里针对小白使用download,当然也可以使用git clone命令)
在这里插入图片描述
下载之后解压放入平时存放代码的目录中:
在这里插入图片描述

2. 配置conda环境

目前官网只针对了conda环境。关于如何安装conda这里就不赘述,请各位同学自行去找相关博客学习安装。
打开anaconda powershell prompt程序,我这里因为装的是miniconda,所以后面带了个miniconda3,这不重要。
在这里插入图片描述
输入命令回车:conda create -n yolov10 python=3.9
`

输入y,继续安装,出现如下信息,表示安装成功:
在这里插入图片描述

输入conda activate yolov10,切换至刚刚创建的yolov10环境
在这里插入图片描述
利用cd命令切换至第一步中下载解压后的源代码所在目录:

cd d:\projects\pycharm\yolov10
请改成你自己的文件夹目录

在这里插入图片描述
然后输入:pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/自动安装所有的依赖库tips:必须先按照上一步切换到下载解压后的源代码所在目录才行
同时会自动根据你是否有gpu自动选择pytorch版本进行按照,这里不需要自己去选择pytorch和cuda按照,非常良心
等待下载库,看个人网速问题,快的话也需要几分钟才行,慢的话可能几个小时
在这里插入图片描述
出现如下信息,则表示下载完成
在这里插入图片描述
继续输入:pip install -e . -i https://pypi.tuna.tsinghua.edu.cn/simple/安装配置环境
出现如下信息表示安装成功
在这里插入图片描述
至此!yolov10的conda虚拟环境搭建全部成功!!

2.训练自己的数据集

请先基于本人前置博客yolov10最详细全面讲解1- 目标检测-准备自己的数据集(yolov5,yolov8均适用)进行!!!!!

2.1训练之前需要先去yolov8官网下载yolov8的预训练权重文件,不然可能训练不了

利用魔法进入yolov8官网
建议将所有yolov8的权重文件都下载下来,以免后续你训练yolov10过程中出现问题。当然也可以根据自己情况单独下载,例如:如果你只用yolov10n那么你只需要下载yolov8n的权重文件,如果你只用yolov10l那么你只需要下载yolov8l的权重文件
在这里插入图片描述
将下载的yolov8的预训练权重文件放入yolov10的文件夹中,如图:
在这里插入图片描述

下载yolov10的预训练权重

进入yolov10官网下载预训练权重,这里以yolov10-n为例
在这里插入图片描述
下载之后放入yolov10代码中新建的weights目录:
在这里插入图片描述

利用命令行命令训练:

yolo detect train data=d:\\modelsdatas\\yolo_datasets\\data.yaml model=yolov10n.pt epochs=5 batch=4 imgsz=640 device=cpu

上述各个参数解释如下,请根据自己的情况修改。

  • yolo detect train:表示进行目标检测的训练,这几个词不需要改
  • data=coco.yaml:指定你自己的数据集yaml文件,这里需要改成我上篇文章中指出的data.yaml文件。
  • model=weights/yolov10n.pt: 指定下载的yolov10预训练权重文件,我这里指定了上述所说。
  • epochs=5:设置训练轮次,可以先设置一个5轮或者10轮,测试看看,顺利进行再设置大一点进行下一步训练。
  • batch=4:设置训练集加载批次,主要是提高训练速度,具体得看你的显卡或者内存容量。如果显存大,则可以设置大一些。或许训练再详细讲解如何设置
  • imgsz=640:设置图片长度,为640像素,这里建议不修改,也可以设置成1280或者其他,会影响训练速度和精度。
  • device=0:指定训练设备,如果没有gpu,则令device=cpu,如果有一个gpu,则令device=0,有两个则device=0,1以此类推进行设置。

自行修改上述参数,输入命令进行训练:成功运行:
在这里插入图片描述
成功运行之后就可以设置epoch=200或者500等进行真正的训练了。

使用代码训练

from ultralytics import yolov10

model = yolov10('yolov10n.pt')
model.train(data='data.yaml', epochs=20, batch=64, imgsz=640,device=[1,2,3,4])

参数意思和上述命令行中的参数一样

当然部分小伙伴可能没有我上述过程这么顺利。如果出现各种报错,大家请在评论区提出,我会针对大家问题一一解答。我将知无不尽,为大家解惑帮助!!还请各位小伙伴多多点赞收藏,支持新人博主,你们的认可是我最大的动力!

(0)

相关文章:

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论

验证码:
Copyright © 2017-2025  代码网 保留所有权利. 粤ICP备2024248653号
站长QQ:2386932994 | 联系邮箱:2386932994@qq.com