我们接着上一篇文章配置完YOLOv5需要的环境后,今天我们试着用YOLOv5训练自己的数据。(在开始本教程前,记得先跑一遍入门篇,确保环境是正常的)
有图有真相,先看看我的运行结果
【YOLOv5 源码地址】
🚀 我的环境:
- 语言环境:Python3.8
- 编译器:PyCharm
- 深度学习环境:
- torch==1.10.0+cu113
- torchvision==0.11.1+cu113
- 显卡:GeForce RTX 3080
- 可供学习使用数据集:🔗水果检测
一、准备好自己的数据
我的目录结构是这样子的