> 文章列表 > YOLO算法改进指南【算法解读篇】:2.如何训练自己的数据集

YOLO算法改进指南【算法解读篇】:2.如何训练自己的数据集

YOLO算法改进指南【算法解读篇】:2.如何训练自己的数据集

我们接着上一篇文章配置完YOLOv5需要的环境后,今天我们试着用YOLOv5训练自己的数据。(在开始本教程前,记得先跑一遍入门篇,确保环境是正常的
有图有真相,先看看我的运行结果


【YOLOv5 源码地址】

? 我的环境:

  • 语言环境:Python3.8
  • 编译器:PyCharm
  • 深度学习环境:
    • torch==1.10.0+cu113
    • torchvision==0.11.1+cu113
  • 显卡:GeForce RTX 3080
  • 可供学习使用数据集:?水果检测

一、准备好自己的数据

我的目录结构是这样子的

  • 主目录