Laser & Optoelectronics Progress, Volume. 58, Issue 8, 0810019(2021)

Mask-Wearing Detection Method Based on YOLO-Mask

Chengshuo Cao and Jie Yuan*
Author Affiliations
  • School of Electrical Engineering, Xinjiang University, Urumchi, Xinjiang 830047, China
  • show less

    Under normalized epidemic prevention and control, mask-wearing detection can promptly remind people to wear masks correctly, thus reducing the risk of cross-infection of people in public places. Aiming at the difficulty of detecting obscured and small targets in the mask-wearing detection task, a YOLO-Mask algorithm is proposed. The proposed algorithm is based on YOLOv3. It introduces an attention mechanism into the feature extraction network to enhance the model's ability to express salient features. Moreover, it uses feature pyramid and path aggregation strategies for feature fusion to enhance detailed feature information and utilize different levels of feature information. The loss function is optimized. The experimental results show that the average accuracy of the YOLO-Mask algorithm is 93.33% for mask-wearing detection targets in different scenarios, which is 7.62% higher than that of the existing YOLOv3 algorithm. The proposed algorithm has better detection results and robustness compared with other mainstream algorithms.

    Tools

    Get Citation

    Copy Citation Text

    Chengshuo Cao, Jie Yuan. Mask-Wearing Detection Method Based on YOLO-Mask[J]. Laser & Optoelectronics Progress, 2021, 58(8): 0810019

    Download Citation

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

    Category: Image Processing

    Received: Aug. 10, 2020

    Accepted: Sep. 20, 2020

    Published Online: Apr. 12, 2021

    The Author Email: Yuan Jie (yuanjie222@126.com)

    DOI:10.3788/LOP202158.0810019

    Topics