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

    科技2025-06-24  12

    流程

    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文件

    通过上图,主要更改上面的路径即可。 1、

    2、

    3、runs文件中exp文件可以全部删除。跑代码会自动生成。

    版本要求

    torch 1.6 torchvision 0.7 如果中间缺少什么包,就补充什么包就行。另外,

    pip install -U requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

    出现错误

    1、出现keyError错误 参考链接:https://blog.csdn.net/weixin_41010198/article/details/108239897 把images和label目录下的,除了图片数据集和标签文件外的其他文件统统删除。 2、需要Xmanager软件来处理X11转发需求

    参考链接:https://www.xshellcn.com/xsh_column/jiaocheng-x11zf.html 下载Xmanager即可解决,甚至不需要上述链接的操作步骤。

    3、可能会出现的问题:版本问题

    最开始运行的是yolov5 的1.0或者是2.0的版本,使用的框架高于要求的框架,会出现

    torch.nn.modules.module.ModuleAttributeError: ‘BatchNorm2d‘ object has no attribute 的错误。 之前还以为是代码有问题,所以真的要多听听他人的建议,不要一意孤行。也是因为代码看不懂,才取消了改代码的想法。hhhh,卑微。 如果出现上述错误,可以参考: https://shliang.blog.csdn.net/article/details/108219810 https://shliang.blog.csdn.net/article/details/108124252

    Processed: 0.011, SQL: 8