前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >基于YOLOv5的NEU-DET钢材表面缺陷任务,加入CFPNet、动态卷积ODConv、多个检测头提升精度

基于YOLOv5的NEU-DET钢材表面缺陷任务,加入CFPNet、动态卷积ODConv、多个检测头提升精度

原创
作者头像
AI小怪兽
发布2023-11-30 15:59:24
4760
发布2023-11-30 15:59:24
举报
文章被收录于专栏:YOLO大作战

1.钢铁缺陷数据集介绍

NEU-DET钢材表面缺陷共有六大类,分别为:'crazing','inclusion','patches','pitted_surface','rolled-in_scale','scratches'

每个类别分布为:

训练结果如下:

2.基于yolov5s的训练

map值:

2.1四个检测头训练结果

map从原始的0.742提升到0.786

2.2 加入即插即用的动态卷积ODConv

https://blog.csdn.net/m0_63774211/article/details/129571319

即插即用的动态卷积ODConv

map从原始的0.742提升到0.761

2.3 ECVBlock

提出的EVC主要由两个并行连接的块组成,其中使用轻量级MLP来捕获顶级特征的全局长期依赖性(即全局信息)。

如何将ECVBlock应用到yolov5/yolov7是本文的关键,重点是增强用于这些检测器的特征金字塔的表示。

1)将ECVBlock添加到backbone或者是head在不同数据集的性能会不一致,比如本文添加到backbone,在NEU-DET钢材表面缺陷和道路缺陷如任务中取得的涨点也是不一样的;

2)比如在backbone添加的位置不同对最终的性能也是完全不一样的,这点也佐证了深度学习具有玄学,体现了调参的必要性,在不断的调参中自然会取得一定经验值;

map从原始的0.742提升到0.758

by CSDN AI小怪兽 https://blog.csdn.net/m0_63774211/article/details/121408804

我正在参与2023腾讯技术创作特训营第三期有奖征文,组队打卡瓜分大奖!

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 1.钢铁缺陷数据集介绍
  • 2.基于yolov5s的训练
    • 2.1四个检测头训练结果
      • 2.2 加入即插即用的动态卷积ODConv
        • 2.3 ECVBlock
        领券
        问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档