首页 > 编程学习 > YOLO算法改进指南【算法解读篇】:2.如何训练自己的数据集

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


【YOLOv5 源码地址】

🚀 我的环境:

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

一、准备好自己的数据

我的目录结构是这样子的

  • 主目录
Copyright © 2010-2022 mfbz.cn 版权所有 |关于我们| 联系方式|豫ICP备15888888号