最近、国内のCOVID-19の流行状況は深刻ですが、一部の公共の場所では、マスクを着用していない、またはマスクを誤って着用している人がいます。そのため、関係スタッフはマスクを正しく着用するように即座に思い出させ、監督する必要があります。しかし、このような重要かつ複雑な作業に直面して、公共の場で自動マスク着用検出を実行する必要があります。この論文では、改良されたYOLOv4に基づく新しいマスク着用検出方法を提案します。具体的には、まず、Coordinate Attention モジュールをバックボーンに追加して、機能の融合と表現を調整します。次に、一連のネットワーク構造の改善を行い、モデルのパフォーマンスと堅牢性を高めます。 3 番目に、K-means クラスタリング アルゴリズムを展開して、9 つのアンカー ボックスを NPMD データセットにより適したものにします。実験結果は、改善された YOLOv4 のパフォーマンスが向上し、ベースラインを 4.06% AP 超え、64.37 FPS の同等の速度であることを示しています。
Recently, the domestic COVID-19 epidemic situation has been serious, but in some public places, some people do not wear masks or wear masks incorrectly, which requires the relevant staff to instantly remind and supervise them to wear masks correctly. However, in the face of such important and complicated work, it is necessary to carry out automated mask wearing detection in public places. This paper proposes a new mask wearing detection method based on the improved YOLOv4. Specifically, firstly, we add the Coordinate Attention Module to the backbone to coordinate feature fusion and representation. Secondly, we conduct a series of network structural improvements to enhance the model performance and robustness. Thirdly, we deploy the K-means clustering algorithm to make the nine anchor boxes more suitable for our NPMD dataset. The experimental results show that the improved YOLOv4 performs better, exceeding the baseline by 4.06% AP with a comparable speed of 64.37 FPS.