当前位置:   article > 正文

YOLOv10训练,适合小白训练,新手YOLOv10训练自己数据集教程!超简单,超详细!!_ultralytics

ultralytics

YOLOv10训练,适合小白训练,新手YOLOv10训练自己数据集教程!超简单,超详细!!


AI学术叫叫兽在这!家人们,给我遥遥领先!!!

方法一:云服务器教程

一、CSDN文字教程点击这个链接即可

二、B站视频教程,非常详细

详细教学视频戳这里,小白八分钟即可完成!不走任何弯路!

方法二 自建环境

训练四部曲,掌握便可跑!

一、COCO本文数据集训练(标注文本格式为text)


1.1、YOLOv10深度学习环境搭建及安装

1.1.1 环境详细配置

我的显卡配置为:RTX A4000(16GB)
具体环境:
PyTorch 1.8.1
Python 3.8(ubuntu18.04)
Cuda 11.1
依赖库按照requirements.txt文件安装即可。

1.1.2 安装YOLOv8所需要的ultralytics库

安装YOLOv8所需要的ultralytics库,在1.1.1的环境中,执行下述代码:

pip install ultralytics
  • 1

1.1.3 下载YOLOv10代码包

下载YOLOv10代码包至此环境中,执行代码如下:

git clone https://github.com/THU-MIG/yolov10.git
  • 1

自动下载
至此,环境搭建结束!

1.2、准备自己的数据集

1.2.1 coco格式的数据集创建(较为简单,适合新手)

文件夹内容如下:

------coco(文件夹名随便起,不能有中文字符)
---------images
------------train(存放训练集图片)
------------test (存放测试集图片)
------------val (存放验证集图片)
---------tables
------------train(存放训练集标注文件)
------------test (存放训练集标注文件)
------------val (存放训练集标注文件)

1.2.2 新建训练执行简易代码

在YOLOv10包中,新建train.py,代码如下:

from ultralytics import YOLO
 
# Load a model
# yaml会自动下载
model = YOLO("ultralytics/cfg/models/v10/yolov10n.yaml")  # build a new model from scratch
#model = YOLO("d:/Data/yolov8s.pt")  # load a pretrained model (recommended for training)
 
# Train the model
results = model.train(data="coco128.yaml", epochs=100, imgsz=640)
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9

此代码在YOLOv10包的存放位置如下图所示:
在这里插入图片描述

1.2.3 执行训练指令,自动配置权重和原始数据集

在配置好的环境中,执行训练指令,如下,

python train
  • 1

然后自动下载原始数据集与权重。等待下载完成即可。

1.3 替换原始数据集为自己的数据集(详细介绍可关注B站:AI学术叫叫兽)

1.3.1 找到原始的数据集,将其删除,将自己的数据集上传,并更名为coco128(原始数据集的名字)

1.3.2 修改coco128.yaml文件中的名字和类别

路径: ultralytics/ultralytics/cfg/datasets/coco128.yaml
做对应修改即可,如果不会修改,直接看本文1.5 路径修改教程即可!

1.4 训练自己的数据集

再次执行训练命令:

python train
  • 1

即可完成自己的数据集训练。

1.5 yaml文件修改路径教学(重要!必学家人们!)

1.5.1 报错情形

家人们,在数据集训练以及改进模型等情形下,常会遇到修改路径的,导致报错。在此进行阐述,希望能够帮到各位粉丝!
例如下图中遇到的错误:
在这里插入图片描述
即:

FileNotFoundError: 'ultralytics/cfg/models/v10/yolov10S.yaml' does not exist
  • 1

类似错误一般都是路径问题。

1.5.2 错误处理

例如:1. 模型网络配置文件:
读懂路径:ultralytics/cfg/models/v10/yolov10S.yaml
其含义为:ultralytics文件下的cfg文件中的models文件中的v8中的YOLOv10yolov10s.yam文件
2. 数据集路径文件

ultralytics/ultralytics/cfg/datasets/coco128.yaml

  • 1
  • 2

打开该文件,找到路径配置,如下:

path: ../datasets/coco128  # dataset root dir
train: images/train2017  # train images (relative to 'path') 128 images
val: images/train2017  # val images (relative to 'path') 128 images
test:  # test images (optional)

  • 1
  • 2
  • 3
  • 4
  • 5

path表示在根目录下的datasets文件中的coco128 文件夹下
train的路径是 images文件中的train2017 文件夹
其它类似。
对应自己的数据或文件路径进行增加、减少路径或修改名称即可。

总结

对于小白,文字教程总是显得苍白无力,对此我在B站上传了相关的视频详解。请关注B站博主:AI学术叫叫兽
更多SCI写作技巧和创新点以及YOLO模型改进源代码等资源,均在B站上传。点开博主主页,在工房中!

AI学术叫叫兽在这!家人们,给我遥遥领先!!!

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Li_阴宅/article/detail/1005851
推荐阅读
相关标签
  

闽ICP备14008679号