Welcome to Journal of Graphics share: 

Journal of Graphics ›› 2022, Vol. 43 ›› Issue (5): 803-814.DOI: 10.11996/JG.j.2095-302X.2022050803

• Image Processing and Computer Vision • Previous Articles     Next Articles

Improved YOLOX method for detecting surface defects of drug blister aluminum foil 

  

  1. 1. College of Control and Computer Engineering, North China Electric Power University, Beijing 102206, China; 

    2. Institute of Computing Technology, Chinese Academy of Sciences, Beijing 100190, China; 

    3. Taicang Institute of Information Technology, Taicang Jiangsu 215400, China

  • Online:2022-10-31 Published:2022-10-28
  • Supported by:
    National Natural Science Foundation of China (61972379); National Key Research and Development Plan Subject (2020YFB1710400)

Abstract:

The surface of aluminum foil in drug blister packaging contains various information on fonts and patterns, and the surface of aluminum foil is uneven, leading to the uneven distribution of light and dark. To address the problem that the YOLOX model cannot more accurately distinguish the defect features from the surface features of aluminum foil, a surface defect detection method based on the improved YOLOX model was proposed. Firstly, in order to enhance the globality of the information input to the Prediction, it was necessary to analyze the global features of the feature map in the Neck network, so the CSP module of the Neck network was replaced with the transformer encoder module. At the same time, the YOLOX model has a deep depth, and to effectively improve the classification accuracy, the Mish activation function was utilized to replace the Swish activation function. Then, focal loss was introduced into the loss function to solve the problem of difficulty in classifying defect regions and background regions due to the similarity of defect features and aluminum foil surface features. The experimental results show that the mAP of the improved model for the detection of aluminum foil surface defects was 90.17%, which was 4.95% higher than the original YOLOX model, and that the improved model can reduce the probability of false detection and missed detection of defects with high similarity to the surface features of aluminum foil.

Key words: aluminum foil surface, defect detection, YOLOX, transformer encoder, Mish, focal loss

CLC Number: