一种带钢表面缺陷分类方法

    公开(公告)号:CN115393646A

    公开(公告)日:2022-11-25

    申请号:CN202211056662.1

    申请日:2022-08-31

    Applicant: 南通大学

    Abstract: 本申请提供一种带钢表面缺陷分类方法,包括:基于第一图像数据集,获取第一缺陷图样;基于Wasserstein距离和第一缺陷图样,构建生成式对抗网络的生成器和判别器,获取第二图像数据集;基于第二图像数据集,构建嵌入注意力模块的Optimal‑ResNet34模型并对Optimal‑ResNet34模型进行训练,获取最优权重;基于最优权重,对第二图像数据集中的第二缺陷图样进行分类以及验证。本申请通过构建生成式对抗网络的生成器和判别器,增加缺陷图像的数量,丰富缺陷特征多样性,解决工业环境下热轧带钢表面缺陷图像匮乏而导致的过拟合问题;其次,将注意力模块引入Optimal‑ResNet34模型中,获取最优权重,能够提高Optimal‑ResNet34模型的分类准确率。

Patent Agency Ranking