YOLOv5基础知识点——网络架构与组件

YOLOv5作者没有分neck模块

Backbone:Focus,BottleneckCSP,SPP

Head:PANet+Detect(YOLOv3/v4 Head)

网络可视化工具:netron

 导出ONNX文件

 

depth-mutiple、width-multiple(s<m<l<x) 

 

anchors——先验框的配置

 三个不同尺度的先验框

[10,13,16,30,33,23] #P3/8——stride=8情况下第一个先验框宽度是10,高度是13,第二个先验框宽度16,高度30.......

一句话CNN:如何理解stride的意义和算法 - 知乎 (zhihu.com)

(17条消息) yolov5增加小目标检测层_AI视觉网奇的博客-CSDN博客_yolov5小目标

Focus

把数据切分为4份,每份数据都是相当于2倍下采样得到的,然后再channel维度(通道维度)进行拼接,最后进行卷积操作

 举例:

 Focus()模块是用来减少浮点运算量,提高速度,但对mAP精度没有帮助

输入经过focus变换: [b,c,h,w]-->[b,c*4,h//2,w//2]

把宽度w和高度h的信息整合到c空间

以YOLOv5s为例,原始640*640*3的图像输入Focus结构,采用切片操作,先变成320*320*12的特征图,在经过一次32个卷积核的卷积操作,最终变成320*320*32的特征图。

而YOLOv5m的Focus结构中的卷积操作使用了48个卷积核,因此Focus结构后的特征图变成320*320*48.YOLOv5l,YOLOv5x也是同样的原理

CSPNet

CSP(Cross Stage Partial Network)跨阶段局部网络

SPP(Spatial Pyramid Pooling)空间金字塔池化

 PANet (Path-Aggregation Network)路径聚合网络

 

 

最适合新手入门的【YOLOV5目标实战】教程!基于Pytorch搭建YOLOV5目标检测平台!环境部署+项目实战(深度学习/计算机视觉)_哔哩哔哩_bilibili

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
基于yolov5的烟火检测可以分为两个主要步骤:模型训练和C++实现部署。 首先是模型训练。我们需要收集包含烟火的图像数据集,并进行数据标注,即标记图像中的烟火目标。接着,在yolov5的基础上进行一些调整和优化,以适应烟火检测任务。然后,使用标注好的数据集进行模型训练,通过迭代优化模型的权重和参数,使其具备较高的烟火检测精度。可以使用GPU来加速模型训练过程。最后,通过评估指标如mAP(平均精度均值)等来评估模型的性能。 其次是C++实现部署。在完成模型训练后,我们需要将其部署到实际环境中进行使用。首先,将模型保存为.onnx或.torchscript格式,以便于在C++中加载和使用。然后,使用C++编程语言,利用相应的深度学习架(如OpenCV、TensorRT等)进行模型加载和推理。通过读取输入图像数据,将其输入到模型中进行烟火检测,获取检测结果。最后,根据实际需求,可以将检测结果进行可视化展示、保存或者与其他业务系统进行集成。 总之,基于yolov5的烟火检测涉及模型训练和C++实现部署两个主要步骤。模型训练需要进行数据收集、标注和调优,以及模型训练和性能评估;C++实现部署需要将模型保存,并利用C++编程语言和深度学习架进行模型加载、推理和结果处理。这样可以实现高效准确的烟火检测功能。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

MUTA️

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值