基于改进RetinaNet的白酒瓶盖缺陷检测方法

作     者：车璇 朱文忠 李韬 胡友鑫 Che Xuan;Zhu Wenzhong;Li Tao;Hu Youxin

作者机构：四川轻化工大学计算机科学与工程学院宜宾644000 

出 版 物：《国外电子测量技术》 (Foreign Electronic Measurement Technology)

年 卷 期：2023年第42卷第4期

页      面：173-180页

学科分类：08[工学] 080203[工学-机械设计及理论] 0802[工学-机械工程] 

基　　金：四川省科技研发重点项目(2019YFG0200) 四川省科技创新(苗子工程)培育项目(2022049)资助 

主　　题：瑕疵检测 RetinaNet Swin Transformer NAS-FPN Soft-NMS 

摘      要：针对瓶装白酒包装质检存在的检测准确度低,小目标重合度高导致误检漏检的情况,提出一种基于RetinaNet的目标检测优化算法,主要使用白酒瓶盖瑕疵数据集进行检测。方法将网络Backbone替换为Swin Transformer,其包含的窗口注意力机制运算有效提升瓶盖瑕疵检测精度同时降低复杂度节省了计算量。在Neck阶段使用神经架构搜索特征金字塔网络(FPN)代替FPN,利用自动架构搜索选出最佳特征融合层,为后续检测提供更高质量的模型,最后采用Soft-NMS降低检测框置信度保留一定真实框,有效的防止瓶盖瑕疵过近或重叠造成漏检。实验证明,改进算法能够精准的识别出各类瓶盖瑕疵,检测精度在白酒瓶盖瑕疵数据集达到了93.53%,相较于原网络提升了8.02%。