运行yolo v5 全过程以及错误出现

本文详细介绍了运行YOLOv5的全过程,包括数据集更改、训练和测试步骤。在训练过程中,重点讲解了如何修改train.py文件,并针对可能出现的关键错误如keyError、版本不匹配等问题提供了解决方案和参考资料。
摘要由CSDN通过智能技术生成

流程

1、测试项目。最开始运行时,需要使用别人的权重,运行python detect.py文件,测试这个项目是否能够正常运行。
2、更改数据集。将data中数据集进行更改,images中放入自己的数据。labels中放入与数据一一对应的标签,保证数据与标签是一一对应的关系。
3、训练数据。运行python train.py文件,会在weights权重文件夹中,或者是runs文件中的最后一个文件中生成一个weights文件夹,如下图所示。
在这里插入图片描述
如果在项目下的weights文件中生成,就不再复制粘贴,如果是在runs文件夹中生成,就复制到项目yolov5-master下的weights文件中,将上图的best.pt覆盖weights中best.pt文件。如下图所示。

在这里插入图片描述
4、测试数据。更改inference中的images数据集。如下图所示,output文件夹保持为空即可。这个文件夹主要用来存放自动打标签的测试数据。
在这里插入图片描述

更改文件

修改train.py文件</
评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值