yolov5训练最常见错误解决办法

我主要用过两种代码

第一种为哔哩哔哩上的up主Bubbliiiing上传的代码———出现以下错误:

1、标注好自己的数据集之后,进行数据集的划分,通常比例为9:1;

2、接下来进行train.py,一般在改完cls_classes.txt之后基本就可以跑起来了,但是在遇到以下问题时“”一般就是因为你的数据集太少,而且还没有改train.py中的冻结阶段训练参数以及解冻阶段两部分的参数设置;比如我的是121张图片,原始代码的参数为:

但是由于我的数据集太少,所以会出现数据集太少,无法训练;因此我又重新设置了训练次数以及batch_size,主要是为了考虑电脑显存,我改了如下:

此时代码可以进行训练,Epoch为50;

3、运行时会出现运行一代或者几代时会运行中断,如以下的错误:

主要时因为我解决之后才想起来发一下,帮助其他有同样错误的伙伴,所以这个没有截当时pycharm中的图,大家可以对比是不是一样的错;

解决措施:

在代码运行的最前方加上 

import os
os.environ['KMP_DUPLICATE_LIB_OK'] = 'True'

刷新后即可运行训练自己的数据集;

4、接着又出现了这个错误:

“OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized”

解决措施:这个错先进行在电脑上检索“libiomp5md.dll”,一般会有两个直接删除时间靠前的即可,如果还是不行就在回收站中进行回收,接着删除另外一个,我的是直接删除就好了的;

 第二种为yolov5官网上的代码,ultralytics/yolov5 at v5.0 (github.com)进行运行时会出现以下错误:

1、会报错:UnicodeDecodeError: 'gbk' codec can't decode byte 0x8b in position 233: illegal multibyte sequence

主要就是因为字节编码解码时两个方式不一样,所以要进行统一规定;

解决措施:在运行的reader.py中检索with open(检索快捷键为ctrl+f),将检索出的所有with open中都加上encoding='utf-8',刷新即可解决;

2、完成上一步改错后,在运行又会出现

 目前这个错还没有解决,咨询说是因为字典问题,有了解的伙伴希望救救孩子!!! 

 希望对大家有帮助

物联沃分享整理
物联沃-IOTWORD物联网 » yolov5训练最常见错误解决办法

发表评论