Computer Science ›› 2024, Vol. 51 ›› Issue (11A): 240100162-7.doi: 10.11896/jsjkx.240100162

• Image Processing & Multimedia Technology • Previous Articles     Next Articles

Graphical LCD Pixel Defect Detection Algorithm Based on Improved YOLOV8

ZHANG Feng   

  1. School of Software,Beihang University,Beijing 100083,China
  • Online:2024-11-16 Published:2024-11-13
  • About author:ZHANG Feng,born in 1985,postgra-duate.His main research interest is production line test system automation and intelligence.

Abstract: During the inspection process of industrial instrument LCD displays,pixel defects are difficult to detect due to its small pixel size.Traditional computer vision methods are sensitive to environmental changes and require manual setting of parameters.In response to the above problems,this paper designs an LCD screen defect detection algorithm based on deep learning,which can identify pixel-level pixel defects on the LCD screen under lower computing power.The main work includes:(1)Aiming at the problem of the small number of positive samples in the sample assigner process of positive and negative samples for small-sized targets,an adaptive positive samples enhancement method for targets of different sizes is proposed.(2)Aiming at the problem of difficulty in small-sized targets training caused by small IoU of positive samples,an adaptive positive sample IoU compensation weighting method is proposed.(3)Aiming at the problem that small data sets are sensitive to hyperparameters in the loss function,a positive and negative cross-entropy imbalance weight classification loss function is designed.(4)In order to solve the pro-blem that detailed features of small-sized targetare difficult to extract,frequency channel attention is introduced in the backbone network to enhance the ability to extract detailed features of small targets.Experiments show that compared with the baselinecomparison model YOLOV8,the mAP_s reaches 63.3%,which is 3.7% higher than the baseline.The mAP_s for pixel defects reaches 78.85%,which improves 4.5%.Meanwhile,the recall rate of pixel defects reaches 99.8%.The mAP_s for dust detection targets reaches 47.8%,improves 3%.These fully verify the effectiveness of the proposed algorithm.

Key words: Minor targets, IoU compensation, Imbalanced weighted loss, Positive samples enhancement

CLC Number: 

  • TP391.41
[1]MING W,ZHANG S,LIU X,et al.Survey of Mura Defect Detection in Liquid Crystal Displays Based on Machine Vision[J].Crystals,2021,11(12):1444.
[2]QIAN J D,CHEN B,QIAN J Y,et al.Fast detection of sub-pixeldefects in LCD based on improved gradient algorithm[J].Journal of Computer Applications,2017,37(S1):201-205.
[3]LUO Q Y.Design and Implementation of Screen Surface DefectDetection System Based on Improved Faster RCNN[D].Beijing:BUPT,2021.
[4]GONG H,MU T,LI Q,et al.Swin-Transformer-EnabledYOLOv5 with Attention Mechanism for Small Object Detection on Satellite Images[J].Remote Sensing,2022,14(12):2861.
[5]LIU Y,SHAO Z.NAM:Normalization-based Attention Module[J].arXiv:2111.12419,2021.
[6]ZHU X,LYU S,WANG X,et al.TPH-YOLOv5:ImprovedYOLOv5 Based on Transformer Prediction Head for Object Detection on Drone-captured Scenarios[C]//IEEE/CVF International Conference on Computer Vision.2021:2778-2788.
[7]WOO S,PARK J.CBAM:Convolutional Block Attention Module[C]//European Conference on Computer Vision(ECCV).2018:3-19.
[8]GHIASI G,LIN T Y,LE Q V.DropBlock:A regularizationmethod for convolutional networks[J].Advances in Neural Information Processing Systems,2018,31.
[9]QIN Z,ZHANG P,WU F,et al.FcaNet:Frequency Channel At-tention Networks[C]//IEEECVF International Conference on Computer Vision.2021:783-792.
[10]LYU C,ZHANG W,HUANG H,et al.RTMDet:An Empirical Study of Designing Real-Time Object Detectors[J].arXiv:2212.07784,2022.
[11]ZHANG S,CHI C,YAO Y,et al.Bridging the Gap Between Anchor-Based and Anchor-Free Detection via Adaptive Training Sample Selection[C]//IEEE CVF Conference on Computer Vision and Pattern Recognition.2020:9759-9768.
[12]LI X,WANG W,WU L,et al.Generalized Focal Loss:Learning Qualified and Distributed Bounding Boxes for Dense Object Detection[J].Advances in Neural Information Processing Systems.2020:21002-21012.
[13]ZHANG H,WANG Y,DAYOUB F,et al.VarifocalNet:AnIoU-aware Dense Object Detector[C]//IEEECVF Conference on Computer Vision and Pattern Recognition.2021:8514-8523.
[14]LI S,HE C,LI R,et al.A Dual Weighting Label Assignment Scheme for Object Detection[C]//IEEE CVF Conference on Computer Vision and Pattern Recognition.2022:9387-9396.
[15]TAN M,PANG R,LE Q V.EfficientDet:Scalable and Efficient Object Detection[C]//IEEE CVF Conference on Computer Vision and Pattern Recognition.2020:10781-10790.
[16]WANG Q,WU B,ZHU P,et al.ECA-Net:Efficient ChannelAttention for Deep Convolutional Neural Networks[C]//IEEECVF Conference on Computer Vision and Pattern Recognition.2020:11534-11542.
[17]FENG C,ZHONG Y,GAO Y,et al.TOOD:Task-aligned One-stage Object Detection[C]//IEEE CVF International Confe-rence on Computer Vision(ICCV).2021:3490-3499.
[18]LIU S,HUANG D,WANG Y.Learning Spatial Fusion for Single-Shot Object Detection[J].arXiv:1911.09516,2019.
[19]CARION N,MASSA F,SYNNAEVE G,et al.End-to-End Object Detection with Transformers[C]//European Conference on Computer Vision.2020:213-229.
[20]ZHU X,SU W,LU L,et al.Deformable DETR:DeformableTransformers for End-to-End Object Detection[J].arXiv:2010.04159,2020.
[1] HU Yimin, Qu Guang, WANG Xiabing, ZHANG Jie, LI Jiadong. EO-YOLOX Model for Insulators Detection in Transmission Lines [J]. Computer Science, 2024, 51(11A): 240200107-6.
[2] CHEN Dong, ZHOU Hao, YUAN Guowu, YANG Lingyu, CHENG Qiuyan, REN Ying, MA Yi. Mountain Fire Detection Algorithm of Transmission Line Based on Multi-scale Features and Coordinate Information [J]. Computer Science, 2024, 51(11A): 230900155-7.
[3] WANG Libin, WANG Shumei. Fundus Vascular Image Segmentation Algorithm Based on Attention Mechanism [J]. Computer Science, 2024, 51(11A): 231000003-6.
[4] LUO Yuetong, ZHAO Dongsheng, PENG Jun, DONG Ziqiu. Research and Application of Color Mapping Function Optimization Method for SegmentedPipeline AxialData Visualization [J]. Computer Science, 2024, 51(11A): 240400039-4.
[5] REN Yuheng, ZHAO Yunfeng, WU Chuang. Deep Gait Recognition Network Based on Relative Position Encoding Transformer [J]. Computer Science, 2024, 51(11A): 240400064-6.
[6] LYU Jiahao, LIU Jinfeng. Domain Generalization and Long-tailed Learning Based on Causal Relationships [J]. Computer Science, 2024, 51(11A): 240300041-8.
[7] XU Jun, ZHOU Peijin, ZHANG Haijing, ZHANG Hao, XU Yuzhong. Analysis of User Evaluation Indicator for AIGC Digital Illustration Design Principles [J]. Computer Science, 2024, 51(11): 47-53.
[8] XIA Shufang, YUAN Bin, QU Zhong. Crack Detection of Concrete Pavement Based on Attention Mechanism and Deep Feature Optimization [J]. Computer Science, 2024, 51(11): 198-204.
[9] LI Xin, PU Yuanyuan, ZHAO Zhengpeng, LI Yupan, XU Dan. Image Arbitrary Style Transfer via Artistic Aesthetic Enhancement [J]. Computer Science, 2024, 51(9): 129-139.
[10] WANG Xiaojie, LIU Jinhua, LU Shuyi, ZHOU Yuanfeng. Color Transfer Method for Unpaired Medical Images Based on Color Flow Model [J]. Computer Science, 2024, 51(8): 176-182.
[11] SHI Dianxi, GAO Yunqi, SONG Linna, LIU Zhe, ZHOU Chenlei, CHEN Ying. Deep-Init:Non Joint Initialization Method for Visual Inertial Odometry Based on Deep Learning [J]. Computer Science, 2024, 51(7): 327-336.
[12] ZENG Peiyi. Classification of Multiscale Steel Microstructure Images Based on Incremental Learning [J]. Computer Science, 2024, 51(6A): 230500180-8.
[13] SU Ruqi, BIAN Xiong, ZHU Songhao. Few-shot Images Classification Based on Clustering Optimization Learning [J]. Computer Science, 2024, 51(6A): 230300227-7.
[14] SUN Ziwen, YUAN Guanglin, LI Congli, QIN Xiaoyan, ZHU Hong. Object Tracking of Structured SVM Based on DIoU Loss and Smoothness Constraints [J]. Computer Science, 2024, 51(6A): 230700113-8.
[15] HUANG Yuanhang, BIAN Shan, WANG Chuntao. Gaussian Enhancement Module for Reinforcing High-frequency Details in Camera ModelIdentification [J]. Computer Science, 2024, 51(6A): 230700125-5.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!