Research on Polarizer Defect Recognition Based on Improved YOLOv5 Algorithm

偏振片作为TFT-LCD的关键部件之一，其表面缺陷严重影响液晶显示器的成像质量。为了实现偏振片缺陷智能化在线检测从而替代目前因人眼检测导致效率低的问题，提出了一种改进型YOLOv5检测算法，即在Backbone层增加了CBAM注意力机制；在Prediction层增加了一个新的输出层；将传统边框回归损失函数改为CIOU_Loss。通过缺陷样本测试实验表明，改进型算法尽管增加了一个输出层导致参数增加，且FPS略微降低，但mAP却提升了4个百分点，并且检测的最高置信度达到了0.93。故改进型YOLOv5算法增强了缺陷目标识别精度和准确度。Polarizer is one of the key components of TFT-LCD. Its surface defects seriously affect the imaging quality of LCD. In order to realize the intelligent on-line detection of polarizer defects and replace the current problem of low efficiency caused by human eye detection, an improved YOLOv5 detection algorithm is proposed, that is, CBAM attention mechanism is added in the backbone layer; A new output layer is added in the prediction layer; The traditional border regression loss function is changed to CIOU_Loss. The defect sample test experiment shows that although the improved algorithm adds an output layer, resulting in an increase in parameters and a slight decrease in FPS, the map increases by 4 percentage points, and the highest confidence of detection reaches 0.93. Therefore, the improved YOLOv5 algorithm enhances the accuracy and accuracy of defect target recognition.