YOLOV5 参数设定与模型训练的坑点一二三

一起养成写作习惯!这是我参与「日新计划 4 月更文挑战」的第10天,点击查看活动详情。

前言

蓝桥杯干完了,怎么说,出题组换了,不讲武德,当然还是自己不行…. Ok,该回来办点正事了。

上次我python怎么读们简单地玩了一下YOLOV5 如果是按照那个我上次记录的文章来走的话,我想,至少大家的YOLO环境应该是搭建好字体识别了,缓存是什么意思也会怎么设定数据集自己训练了。然后我说我在玩的时候,说我翻车了,说图片尺寸有问题,其实呢,字体天下我没有翻车,没错,没有翻车,是我在测试,模型训练的时候参数的问题。为什么我会说我自己没有翻车咧。很简单。

YOLOV5 参数设定与模型训练的坑点一二三

但是为什么会出不来结果呢,这里面的原因有很多,最容易出现,也最有可能的原因就是置信度的问题。那么今天python怎么读我要搞定的问题就是让我自己的模型出来结果。缓存并且做一个记录,

自训练模型问题

无目标框

问题缓存视频合并app分析

这个就是咱们现在遇到的问题,那就是模型明明训练好了,但是却没有目标框出现。而且在做测试的时候,我们使用可是同一个训练集里面的图片。并且在我们实际上的训练过程当中也发现它其实应该是已经识别出来了。但是就是没有框出来。

先说可能的缓存的视频怎么保存到本地原因,第一个也机器学习是比较容易误导人的就是这个玩意

detect.py

YOLOV5 参数设定与模型训练的坑点一二三
train.py

这个玩意其实根本就不是指你图片的原始大小。而是送入神字体美化大师经网络的时候会被处理的大小,之说以需要这样,如果看了我前面的那篇深度学习Pytorch 从0到部署 应该是知道的,这里我不想复述,看着片文章字体天下的多少的多少都是有基础的。

所以,我前面是被误导了。那竟然不是这个原因还能是什么原因呢。显然这里还有一个原因那就是 可信度。

问题来了,如果是可信度的问题,那么为什么训练测试的就好好的有明显的的目标标深度学习注呢。这个怎么说呢。深度学习本质上是一个机器学习。是一个学习的过程,并且神经网络python123的训练本质上应该是一个建模拟合的过程。

我们举一个最直观的例子,那就是学生学习做试卷。

一个学生想要取得好成绩,无非两条件字体识别,天赋+学习。

在我们这里 天赋就是游戏的神经网络结构,对神经网络进行优化。学习自然就是获取权重的过程,也就是训练过程。

我们在排查问题的现在,显然我们先排除天赋问题,也就是神经网络结构的问题。 所以现在的问题来到了学习上。

学习有多种方式,要么题海战术,要么精打细算(认真搞清楚会的) 所以这里引出了我们深度学习需要的几个东西:数据量(题目),学习速率,学习时间(学习轮数)。

那么我们看看题海战术会出现什么问题,首先采用这个策略需要有足够的数据集,但是这里有个问题,你题目做的多,看的题型多,并不意味着,你就会了呀,做的多不代表懂得多。深度学习的过程,相当于拿着标准答案去刷题(所以字体大全你测试的时候能够出来很正常)。但是实际上考试的时候,没有答案,你见到题目多但是,你没有掌握到精髓呀,所以自然题目做不出来,那么也就是识别不出来。字体天下

看看第二个,精打细算。这个属于什么呢,字体不拼数量,拼精度。学会举一反三,问题就是学得慢,适合咱们数据集小的情况。 也就是咱们现在的情况,一共才24张图片。所以,这个时候我们要干嘛,提高精度呀,学久一点,学慢一点,慢慢来。

那么实际上控制这些的有哪些参数呢

我们把目光放在train.py 这个文件

YOLOV5 参数设定与模型训练的坑点一二三
当然还有一个 lr 这个缓存英文没必要动,我们也可以通过提高学习轮数来达到效缓存视频合并app下载果。

第一个不用说,学久一点,第二个 batch-size

这个参数很重要,两个原因,字体识别扫一扫第一个我们是在显卡上面跑,这个的大小涉及到我们显卡性能 第二个,这个大小也就是我说字体大小怎么调的,你要不要学精的问题,越大,学得越快,N个字体大全题目一块看,自然学不到啥,当然是在相同的学习时长(轮数)下,设置小一点,例如这里是4(默认是8)那就一次性学4个题目,自然会精度高一机器学习点。

重新训练

来看看咱们原来的指令

python train.py --weights weights/yolov5s.pt  --cfg models/yolov5s.yaml  --data data/mydata.yaml --epoch 200 --batch-size 4   --device 0

首先第字体设计一个呢,其实迁移学习的意思,用训练好的参数去再训字体练,而不是默认值,这样会快一点 第二个参数呢,是那个模型的参数设置,分几个类呀啥字体美化大师的 第三个参数,指定咱们的数据集 第四个参数,训练次数(时长,这样理解) 第五个参数,一次性进入训练的数据集大小(个数) 第六个参数,训python编程练平台(我这里是cuda0也就是用我自己的显卡训练GTX1650(害,穷,垃圾卡先用着))

所以现在,我们训练的轮数可能太少了python怎么读,所以嘞,我们就用300看看(默认也是300)

现在这里的话为了方便,我就直接在文件里字体下载面改了

YOLOV5 参数设定与模型训练的坑点一二三

直接让它运行,之后咱们来查看结果

tensorboard --logdir=runs

YOLOV5 参数设定与模型训练的坑点一二三
YOLOV5 参数设定与模型训练的坑点一二三

测试 现在咱们测字体大全试一下咱们的模型 同样的为了方便,我这里也是直接该参数在文件里面搞

YOLOV5 参数设定与模型训练的坑点一二三
这里的话,一方面是为了方便,还有就是有些超参数我懒得去设置了。 来看看效果:

YOLOV5 参数设定与模型训练的坑点一二三

好框出来了,但是问题来了 问号是什么玩意

中文乱码

好不容易解决了训练问题,接下来是这个毛病,这个呢也好解决,主要是opencv的问题。

这玩意默认是不支持中文的,在我们当前的这个版本当中,问题定位到这里。 所以修改一下源码就好了。

YOLOV5 参数设定与模型训练的坑点一二三

一开始 我的解决方案是直接在Puttext指定那个字体,但是可能是和我一般使用的cv版本不一样,没有设置参数。 所以最后也是在Bing了一下,字体大全之后找到解决方机器学习案。

先来看看咱们解决好之后的样子

YOLOV5 参数设定与模型训练的坑点一二三

先来一个搜索小技巧

yolov5-5 中文框 -csdn

原文参考链接 zhuanlan.zhi缓存视频合并app下载hu.com/p/3594字体管家94157

那么如何解决呢,首先的话得先下载字体

链接:pan.baidu.com/s/1qXVX字体天下BgfD… 提取码:6666

下载之后解压,这里的话我放在了这个文件夹

YOLOV5 参数设定与模型训练的坑点一二三

然后找到缓存视频合并app下载咱们的 这个文件

YOLOV5 参数设定与模型训练的坑点一二三
找到这个函数
YOLOV5 参数设定与模型训练的坑点一二三
修改代码

def plot_one_box(x, img, color=None, label=None, line_thickness=3):
    # Plots one bounding box on image img
    tl = line_thickness or round(0.002 * (img.shape[0] + img.shape[1]) / 2) + 1  # line/font thickness
    color = color or [random.randint(0, 255) for _ in range(3)]
    c1, c2 = (int(x[0]), int(x[1])), (int(x[2]), int(x[3]))
    cv2.rectangle(img, c1, c2, color, thickness=tl, lineType=cv2.LINE_AA)
    if label:
        tf = max(tl - 1, 1)  # font thickness
        t_size = cv2.getTextSize(label, 0, fontScale=tl / 3, thickness=tf)[0]
        font_size = t_size[1]
        font = ImageFont.truetype('../data/font/simsun.ttc', font_size)
        t_size = font.getsize(label)
        c2 = c1[0] + t_size[0], c1[1] - t_size[1]
        cv2.rectangle(img, c1, c2, color, -1, cv2.LINE_AA)  # filled
        img_PIL = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
        draw = ImageDraw.Draw(img_PIL)
        draw.text((c1[0], c2[1] - 2), label, fill=(255, 255, 255), font=font)
        return cv2.cvtColor(np.array(img_PIL), cv2.COLOR_RGB2BGR)
    # if label:
    #     tf = max(tl - 1, 1)  # font thickness
    #     t_size = cv2.getTextSize(label, 0, fontScale=tl / 3, thickness=tf)[0]
    #     c2 = c1[0] + t_size[0], c1[1] - t_size[1] - 3
    #     cv2.rectangle(img, c1, c2, color, -1, cv2.LINE_AA)  # filled
    #     cv2.putText(img, label, (c1[0], c1[1] - 2), 0, tl / 3, [225, 255, 255], thickness=tf, lineType=cv2.LINE_AA)

YOLOV5 参数设定与模型训练的坑点一二三

mosaic = plot_one_box(box, mosaic, label=label, color=color, line_thickness=tl)

回到 detect.py

YOLOV5 参数设定与模型训练的坑点一二三

im0 = plot_one_box(xyxy, im0, label=label, color=colors[int(cls)], line_thickness=3)

ok ,到目前为止就好了。 哦,此外,除了上次编码的错误外,为了安全起见,这个 text.py 下面也可以修改一下

YOLOV5 参数设定与模型训练的坑点一二三

到这里,就OK了。

那么接下来是那些参数的设置。搞清楚这些,差不多就能熟练地使用了。

接下来就是先啃论文,然后怼源码。最后工程化 YOL缓存视频怎样转入相册OV5,然后做项目,没错我就是为了互联网+ 比赛铺路。

参数

detect.py 参数

weights:训练的权重
source:测试数据,可以是图片/视频路径,也可以是'0'(电脑自带摄像头),也可以是rtsp等视频流
output:网络预测之后的图片/视频的保存路径
img-size:网络输入图片大小
conf-thres:置信度阈值
iou-thres:做nms的iou阈值
device:设置设备
view-img:是否展示预测之后的图片/视频,默认False
save-txt:是否将预测的框坐标以txt文件形式保存,默认False
classes:设置只保留某一部分类别,形如0或者0 2 3
agnostic-nms:进行nms是否也去除不同类别之间的框,默认False
augment:推理的时候进行多尺度,翻转等操作(TTA)推理
update:如果为True,则对所有模型进行strip_optimizer操作,去除pt文件中的优化器等信息,默认为False

train.py 参数

weights:训练的权重(这里主要是为了迁移学习,你也可以不要default=‘’)
cfg:模型配置文件,网络结构
data:数据集配置文件,数据集路径,类名等
hyp:超参数文件
epochs:训练总轮次
batch-size:批次大小
img-size:输入图片分辨率大小
rect:是否采用矩形训练,默认False
resume:接着打断训练上次的结果接着训练
nosave:不保存模型,默认False
notest:不进行test,默认False
noautoanchor:不自动调整anchor,默认False
evolve:是否进行超参数进化,默认False
bucket:谷歌云盘bucket,一般不会用到
cache-images:是否提前缓存图片到内存,以加快训练速度,默认False
weights:加载的权重文件
name:数据集名字,如果设置:results.txt to results_name.txt,默认无
device:训练的设备,cpu;0(表示一个gpu设备cuda:0);0,1,2,3(多个gpu设备)
multi-scale:是否进行多尺度训练,默认False
single-cls:数据集是否只有一个类别,默认False
adam:是否使用adam优化器
sync-bn:是否使用跨卡同步BN,在DDP模式使用
local_rank:gpu编号
logdir:存放日志的目录
workers:dataloader的最大worker数量

总结

大概就这么多,主要结合上篇博文的后续补充。

发表评论

提供最优质的资源集合

立即查看 了解详情