基于改进的YOLOv4-tiny钢卷端面缺陷检测

作者:吴奎; 向峰; 周顺; 张雪荣; 李红军; 张驰
更新时间:

针对带钢生产过程中中钢线圈端面缺陷和拉拔缺陷,提出了一种改进的yolov4微检测方法。首先,在主干网中加入注意模块,增强检测模型对缺陷特征的聚焦能力,并加入上采样层模型,优化深度特征和浅层特征的融合。其次,采用Focal损失函数替换置信度和分类的二等分交叉熵损失函数,解决了分类过程中正类和负类之间的不平衡问题。最后,通过加权聚类得到一个新的先验框架。实验结果表明,改进模型的参数和检测速度与原模型相当,但检测精度取得了较好的效果,更适合带钢生产的实时检测任务。