流程
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文件夹保持为空即可。这个文件夹主要用来存放自动打标签的测试数据。