Computer Engineering and Applications ›› 2024, Vol. 60 ›› Issue (2): 295-303.DOI: 10.3778/j.issn.1002-8331.2209-0375

• Engineering and Applications • Previous Articles     Next Articles

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

CHEN Le, ZHOU Yongxia, ZU Jiazhen   

  1. College of Information Engineering, China Jiliang University, Hangzhou 310018, China
  • Online:2024-01-15 Published:2024-01-15

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

陈乐,周永霞,祖佳贞   

  1. 中国计量大学 信息工程学院,杭州 310018

Abstract: 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 surface defect detection algorithm is proposed in this paper. The adaptive balanced feature pyramid (ABFP) module is proposed to sufficiently incorporate the multi-level features extracted by the backbone network, while increasing the detection 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 mAP50 and mAP50: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.

Key words: polarizer surface, defect detection, YOLOX-S, adaptive balance feature pyramid, CIoU, Mish

摘要: 偏光片是液晶显示器的重要组成部分,其表面缺陷不仅会降低液晶显示器的显示质量,甚至可能造成整个液晶面板的报废。针对偏光片表面缺陷存在尺度差异大、形状变化多样的问题,提出一种改进YOLOX-S的偏光片表面缺陷检测算法。提出自适应平衡特征金字塔(ABFP)模块充分融合主干网提取的多级特征,并通过单个卷积增加检测分支,进一步增强模型的多尺度检测能力。在ABFP中引入注意力模块CBAM关注重要特征。采用CIoU损失函数的同时使用Mish激活函数替代SiLU激活函数。实验结果表明,改进的算法在偏光片表面缺陷数据集上的mAP50和mAP50:95分别达到92.97%和55.16%,相比YOLOX-S(FPN)提升了1.86和1.34个百分点,每秒检测帧数(FPS)达到50,基本满足工业实时检测的需求。

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