结合Swin-Transformer的改进YOLOv5s包装盒缺陷检测算法  

Improved YOLOv5s Packaging Defect Detection Algorithm Combined with Swin-Transformer

作　　者：赵敏 范英[1] 高思伟 谢佳泽 王潇 ZHAO Min;FAN Ying;GAO Si-wei;XIE Jia-ze;WANG Xiao

机构地区：[1]太原科技大学车辆与交通工程学院,山西太原030024

出　　处：《制造业自动化》2024年第12期34-40,共7页Manufacturing Automation

基　　金：山西省重点研发计划(201903D121176);山西省研究生教改项目(2022YJJG208);山西省研究生创新项目(2022Y704)。

摘　　要：针对已有目标检测算法在缺陷检测方面识别精度低、泛化能力差的问题,提出一种基于改进YOLOv5s的包装盒缺陷检测方法。首先,针对包装盒缺陷特征复杂难以识别分类的问题,主干特征提取网络CSPDarknet结构替换为Swin-Transformer结构,提高模型对缺陷特征信息的获取能力和检测精度;其次,在预测端加入Dropout和Label smoothing正则化方法,解决破损包装盒图像中不同缺陷特征交叉分布导致模型识别能力差的问题。试验结果表明,改进后YOLOv5s模型检测精度平均精度均值mAP提升了10.7%,测试时,能有效检测出更多的包装盒缺陷。该模型在包装盒缺陷检测识别任务中检测精度提高,误检率和漏检率降低,有效提高了模型泛化能力。

关 键 词：缺陷检测 YOLOv5s Swin-Transformer结构 正则化 

分 类 号：TB487[一般工业技术—包装工程]