1. 改进 YOLO 的口罩佩戴实时检测方法.
- Author
-
程长文, 陈玮, 陈劲宏, and 尹钟
- Subjects
- *
COVID-19 pandemic , *DATA augmentation , *MEDICAL masks , *ALGORITHMS , *CLASSIFICATION - Abstract
The existing YOLO target detection algorithm, based on one-stage concept, aims for multi-objective detection. But this algorithm performs not well for dual classification detection and add more performance overhead while detection. In order to improve the detection efficiency of dual classification mask wearing during the period of COVID-19, this paper proposes a real-time detection method based on YOLO for detecting the condition of bi-objective mask wearing. The optimization is based on the latest YOLO version 4: firstly, the feedforward input layer of the model is improved to optimize the data augment part and the adaptive image scaling is added to improve the efficiency of the detection for both dual classification and small objectives; secondly, the adaptive anchoring frame is added to replace the activation function so as to reduce the computation and thus improve its efficiency; thirdly, the optimization of Neck and the addition of Focus structure improve the capability of feature fusion and reduce the amount of parameters to raise the efficiency. The experimental results showed that, compared with the existing YOLO version 4, using this method in this paper improves F1 by 0.33% and mAp by 0.71% in the data set. Also, the detection efficiency is also improved significantly under the same experimental condition. [ABSTRACT FROM AUTHOR]
- Published
- 2023
- Full Text
- View/download PDF