利索能及
我要发布
收藏
专利号: 2023107746638
申请人: 盐城工学院
专利类型:发明专利
专利状态:已下证
更新日期:2025-12-08
缴费截止日期: 暂无
联系人

摘要:

权利要求书:

1.一种基于改进YOLOv5的带钢表面缺陷检测方法,其特征在于,包括:搭建改进的anchor‑freeYOLOv5网络;

获取带钢表面图像;

将所述带钢表面图像输入至所述anchor‑freeYOLOv5网络,获得带钢表面缺陷检测结果;

输出所述带钢表面缺陷检测结果。

2.如权利要求1所述的一种基于改进YOLOv5的带钢表面缺陷检测方法,其特征在于,所述anchor‑freeYOLOv5网络包括:依次连接的特征图模块、特征融合模块、卷积模块和探测器模块。

3.如权利要求1所述的一种基于改进YOLOv5的带钢表面缺陷检测方法,其特征在于,所述anchor‑freeYOLOv5网络的工作机制包括:依次连接的input网络模块、Backbone网络模块、Neck网络模块和Prediction网络模块。

4.如权利要求3所述的一种基于改进YOLOv5的带钢表面缺陷检测方法,其特征在于,所述input网络模块对输入的所述带钢表面图像进行Mosaic数据增强。

5.如权利要求4所述的一种基于改进YOLOv5的带钢表面缺陷检测方法,其特征在于,所述Backbone网络模块中有用于对Mosaic数据增强后的所述带钢表面图像进行特征图提取的Focus结构、CSPDarknet结构;

所述CSPDarknet结构包含5个CSP模块。

6.如权利要求5所述的一种基于改进YOLOv5的带钢表面缺陷检测方法,其特征在于,所述Neck网络模块采用BiFPN模块+PAN模块结构;

所述Backbone网络模块的输出作为BiFPN模块的输入,进行特征融合,获得特征金字塔;

PAN模块先复制特征金字塔中最底下的一层,变成新特征金字塔的最底层;

将新特征金字塔的最底层进行下采样操作;

将特征金字塔的倒数第二层进行一个3x3的卷积,步幅为2,与下采样后的最底层进行一个横向连接两者相加;相加方式采用concat操作;

最后再来一个3x3卷积来融合相加结果的特征。

7.如权利要求6所述的一种基于改进YOLOv5的带钢表面缺陷检测方法,其特征在于,所述Prediction网络模块输出时使用EIoUloss。

8.如权利要求3所述的一种基于改进YOLOv5的带钢表面缺陷检测方法,其特征在于,所述anchor‑freeYOLOv5网络在训练时采用EIoU作为边界框回归损失。

9.如权利要求8所述的一种基于改进YOLOv5的带钢表面缺陷检测方法,其特征在于,所述Neck网络模块特征融合后,基于Anchor‑free边界框回归使用两个卷积层来预测编码边界框的三维张量、对象性和类预测。

10.一种基于改进YOLOv5的带钢表面缺陷检测方法,其特征在于,包括:搭建模块,用于搭建改进的anchor‑freeYOLOv5网络;

获取模块,用于获取带钢表面图像;

检测模块,用于将所述带钢表面图像输入至所述anchor‑freeYOLOv5网络,获得带钢表面缺陷检测结果;

输出模块,用于输出所述带钢表面缺陷检测结果。