Chinese Journal of Liquid Crystals and Displays, Volume. 38, Issue 8, 1139(2023)

Mask wearing detection based on improved YOLOv7

Hui-chen FU1,2, Jun-wei GAO1,2、*, and Lu-yang CHE1,2
Author Affiliations
  • 1School of Automation, Qingdao University, Qingdao 266071, China
  • 2Shandong Key Laboratory of Industrial Control Technology, Qingdao 266071, China
  • show less

    Wearing masks is an effective way for preventing COVID-19 and cooperating with the national epidemic prevention and control. An improved YOLOv7 algorithm is proposed to solve the problems such as whether masks are correctly worn, different shooting angles and being blocked. Based on YOLOv7, the convolutional attention mechanism is introduced into the Head region of the network to make the feature network more targeted in the processing of the mask region, thus enhancing the learning ability of the feature network to the mask region. The structure of Backbone area is optimized, the ConvNeXt network structure is improved, and partial convolution is introduced into the network instead, which improves the detection accuracy and robustness of the model and enhances the accuracy of prediction without introducing a large number of additional calculations. The space pyramid pool of the Head layer is improved to improve the training speed and accelerate the model convergence. Experiments show that in the case of complexity and occlusion, the loss function of the improved YOLOv7 decreases significantly, and the mAP on the test set is 93.8%, which is 3.6% higher than that of the original YOLOv7 algorithm.The accuracy of each category is improved, and the accuracy of no mask, correct mask and incorrect mask are increased by 6.8%, 2.1% and 1.7%, respectively. The cases of error detection are significantly reduced, and the generalization ability is significantly improved.

    Tools

    Get Citation

    Copy Citation Text

    Hui-chen FU, Jun-wei GAO, Lu-yang CHE. Mask wearing detection based on improved YOLOv7[J]. Chinese Journal of Liquid Crystals and Displays, 2023, 38(8): 1139

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Research Articles

    Received: Nov. 8, 2022

    Accepted: --

    Published Online: Oct. 9, 2023

    The Author Email: Jun-wei GAO (qdgao163@163.com)

    DOI:10.37188/CJLCD.2022-0371

    Topics