The application and research of improved YOLOv10 algorithm in smoking behavior detection on campus
In order to maintain a good living and learning environment and students' health in campus, it is particularly important to detect students' smoking behavior in campus. With the continuous development of artificial intelligence, object detection algorithms based on computer vision are becoming increasingly perfect. However, due to the small size of the cigarette target, existing object detection algorithms have serious missed detection and false detection phenomena, and are difficult to use in a real campus environment. Based on the YOLOv10 object detection algorithm, this paper proposes a multi-level end-to-end detection model M-YOLOv10. Aiming at the limited feature extraction capabilities of the original YOLOv10, a self-attention mechanism is introduced to improve the model's global feature extraction capabilities. At the same time, the feature fusion module is improved to achieve multi-level fusion and improve the model's small object detection capabilities. Experimental results show that compared with the model before improvement, the M-YOLOv10 algorithm improves [email protected] by 17.6%, and the model detection speed remains basically unchanged.
Authors
- Li Lu
Publication Details
- Published
- 2024-10-24
- DOI
- https://doi.org/10.1117/12.3050443
- Primary Topic
- Advanced Computing and Algorithms
- Type
- article
- Field-Weighted Citation Impact
- 0.00