基于改进YOLOv4算法的铝材表面缺陷识别方法研究

• 作者: 栾明慧 李松松 李晨 王宇恒 郭忠宇
• 来源期刊: 现代信息科技
• 学科: 工学
• 关键词: 铝材表面缺陷 YOLOv4 多尺度预测 K-means算法
• 年，卷（期）: 2021,（23）
• 所属期刊栏目: 计算机技术|Computer Technology
• 页码范围: 96-99,103
• 页数: 5页
• 分类号: TP391.4
• 语种: 中文
• DOI: 10.19850/j.cnki.2096-4706.2021.23.025

摘要

文章针对铝材表面缺陷识别原始算法精度低与提取突出特征能力弱的问题,提出一种改进的YOLOv4算法.首先,为提高对小目标缺陷的检测能力,改进了多尺度预测,增强更浅层的细粒度特征信息融合;其次,对铝材标注数据样本采用K-means聚类,获取更适合缺陷目标的先验框.实验结果表明,在检测速度基本不变的前提下,改进YOLOv4算法的平均精度达到95.02％,比原始的YOLOv4算法提高了1.42％,比YOLOv3提高了2.34％,比Faster R-CNN提高了11.48％.