一种基于模型剪枝的缺陷预测方法

    公开(公告)号:CN116342533A

    公开(公告)日:2023-06-27

    申请号:CN202310309856.6

    申请日:2023-03-28

    Abstract: 本发明公开了一种基于模型剪枝的缺陷预测方法,根据缺陷的尺度自适应方法生成对应的三层热力图尺度标签;构建若干个等间隙的膨胀卷积核;将膨胀卷积核组建成一种串联和并联关系上的等间隙膨胀卷积网络;将网络拼接在标准的ResNet50网络后,输出最高层的预测目标,对最高层热力图标签H3进行训练;剪枝掉由等间隙膨胀卷积构成的网络,由ResNet50网络对应特征层分别预测输出的三个尺度和三层热力图标签进行损失梯度下降训练指定轮次。将各层的结果进行相应的后处理后,把各层的结果整合起来,将预测的结果还原到原图上。本申请在串联上相加与并联上拼接而形成全面地结合,在扩大感受野的同时,形成了一种全局和局部信息的交叉获取。

Patent Agency Ranking