改进YOLOX-S的偏光片表面缺陷检测算法  

Surface Defect Detection of Polarizer Based on Improved YOLOX-S Algorithm

作　　者：陈乐 周永霞 祖佳贞 CHEN Le;ZHOU Yongxia;ZU Jiazhen(College of Information Engineering,China Jiliang University,Hangzhou 310018,China)

机构地区：[1]中国计量大学信息工程学院,杭州310018

出　　处：《计算机工程与应用》2024年第2期295-303,共9页Computer Engineering and Applications

基　　金：浙江省自然科学基金(LY19F030013)。

摘　　要：偏光片是液晶显示器的重要组成部分,其表面缺陷不仅会降低液晶显示器的显示质量,甚至可能造成整个液晶面板的报废。针对偏光片表面缺陷存在尺度差异大、形状变化多样的问题,提出一种改进YOLOX-S的偏光片表面缺陷检测算法。提出自适应平衡特征金字塔(ABFP)模块充分融合主干网提取的多级特征,并通过单个卷积增加检测分支,进一步增强模型的多尺度检测能力。在ABFP中引入注意力模块CBAM关注重要特征。采用CIo U损失函数的同时使用Mish激活函数替代Si LU激活函数。实验结果表明,改进的算法在偏光片表面缺陷数据集上的m AP_(50)和m AP_(50:95)分别达到92.97%和55.16%,相比YOLOX-S(FPN)提升了1.86和1.34个百分点,每秒检测帧数(FPS)达到50,基本满足工业实时检测的需求。Polarizers are an essential part of liquid crystal displays,whose surface defects not only reduce the display quality of liquid crystal displays,but also cause the scrapping of the entire liquid crystal panel.Aiming at the problems that the surface defects of polarizers have large scale differences and various shapes,an improved YOLOX-S polarizer sur-face defect detection algorithm is proposed in this paper.The adaptive balanced feature pyramid(ABFP)module is pro-posed to sufficiently incorporate the multi-level features extracted by the backbone network,while increasing the detec-tion branch through a single convolution to further enhance the multi-scale detection capability of the model.Then,an attention module CBAM is introduced in ABFP to focus on important features.In addition,the Mish activation function is used instead of the SiLU activation function while the CIoU loss function is adopted.The experimental results show that the improved algorithm achieves 92.97%and 55.16%of mAP_(50) and mAP_(50:95) on the polarizer surface defect dataset,which are 1.86 and 1.34 percentage points higher than that of YOLOX-S(FPN).Frames per second reaches 50,which basically meets the needs of industrial real-time detection.

关 键 词：偏光片表面 缺陷检测 YOLOX-S 自适应平衡特征金字塔 CIoU Mish 

分 类 号：TP391[自动化与计算机技术—计算机应用技术]